GitHub repository: vsitzmann/pytorch_prototyping
Primary language: Python
Description: Custom pytorch modules with sane default parameters useful for model prototyping. (U-Net, upsampling, downsampling...)