/pytorch_prototyping

Custom pytorch modules with sane default parameters useful for model prototyping. (U-Net, upsampling, downsampling...)

Primary LanguagePythonMIT LicenseMIT

Stargazers