ml4gw.nn.resnet
Modules
In large part lifted from https://github.com/pytorch/vision/blob/main/torchvision/models/resnet.py but with 1d convolutions and arbitrary kernel sizes, and a default norm layer that makes more sense for most GW applications where training-time statistics are entirely arbitrary due to simulations. |
|
In large part lifted from https://github.com/pytorch/vision/blob/main/torchvision/models/resnet.py but with arbitrary kernel sizes |