yash1994:

Don’t make any changes to the source code and implement a custom network and then add weights to your custom model by iterating state_dict of pretrained ResNet50 from torchvision.


But I have to add the attention mechanism codes to chage the network and get the weights, so what can i do to solve it?