search

rdevon / cortex

A machine learning library for PyTorch
BSD 3-Clause "New" or "Revised" License
92 stars 25 forks source link

Added Spectral Norm #21

Closed lavoiems closed 6 years ago

lavoiems commented 6 years ago

Added SpectralNormLayer. SpectralNormLayer is a normalization technique defined here: https://openreview.net/forum?id=B1QRgziT-

The strategy used to implement a wrapper over some pytorch layer. I added an option to use the spectralnorm on SimpleConvnet and added an option to use it gan.

rdevon commented 6 years ago

@IshmaelBelghazi Can you take a quick look at this?

rdevon commented 6 years ago

There's conflicts how due to updates to the framework. Fixing this should be easy. Once this is done, I will do the review.