Pytorch Utils
Algorithms and training
torchelie.nn.utils.
weight_scale
Multiply getattr(module, name) by scale on forward pass as a hook. Used to implement equalized LR for StyleGAN
getattr(module, name)
scale