Correct way to calculate FLOPS in model

[sio277 (shoh)]

Thanks for the nice work.
FLOPS count for the spectral normalization is also supported?
(torch.nn.utils.spectral_norm — PyTorch 1.8.1 documentation)