sample_spectral_norm.py

import torch.nn as nn
from torch.nn.utils import spectral_norm


if __name__ == '__main__':
    layer = nn.Conv2d(3, 15, 3, 1, 1)
    layer = spectral_norm(layer)

    torch.save(layer.state_dict(), 'tmp.pth')

    layer1 = nn.Conv2d(3, 15, 3, 1, 1)
    layer1 = spectral_norm(layer1)

    sd = torch.load('tmp.pth')
    print(list(sd.keys()))
    layer1.load_state_dict(sd)