Hi,
it seems you are still editing source code, don’t add ModifiedResNet50 to resnet.py. You write/implement your custom class in your script/notebook cell. import bottleneck layers from torchvision if needed. Also change super(ResNet, self).init() to super(ModifiedResNet50, self).init(). I can’t write the whole code here, i can suggest a direction (so direct copy paste wont work here), rest you have to implement. Hope you understand.