vsitzmann / pytorch_prototyping

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

Related projects

Alternatives and complementary repositories for pytorch_prototyping