Just would like to know how to refactor 'spectral_normalization.py' such that it applies spectral normalization to the weights of a recurrent layer (e.g., GRU). Is it correct to change the 'name' argument of the init method of the SpectralNorm class so to indicate 'w_ih' and 'w_hh' instead of 'weight'?
Hi,
Just would like to know how to refactor 'spectral_normalization.py' such that it applies spectral normalization to the weights of a recurrent layer (e.g., GRU). Is it correct to change the 'name' argument of the init method of the SpectralNorm class so to indicate 'w_ih' and 'w_hh' instead of 'weight'?
Best