Parametrized version of spectral norm fc and conv

jhjacobsen / invertible-resnet

Official Code for Invertible Residual Networks

MIT License

517 stars 78 forks source link

Parametrized version of spectral norm fc and conv #34

Open balalofernandez opened 1 month ago

balalofernandez commented 1 month ago

Newer versions of pytorch require applying spectral norm through parametrizations, so I added to files that uses torch.nn.utils.parametrizations for spectral norm.