Your approach should work and the torchvision
implementation of ResNet
uses a similar approach as seen here.
1 Like
Your approach should work and the torchvision
implementation of ResNet
uses a similar approach as seen here.