grau4 / pytorch_prototyping Goto Github PK
View Code? Open in Web Editor NEWThis project forked from vsitzmann/pytorch_prototyping
Custom pytorch modules with sane default parameters useful for model prototyping. (U-Net, upsampling, downsampling...)
License: MIT License