Hello guys,
I follow this tutorial:
http://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html#convnet-as-fixed-feature-extractor 35
to apply resnet on CIFAR10. But when I ran my model, I got an error:

image.png1011×53 8 KB

The model stop right at AvgPool layer

image.png824×306 29.8 KB

The error came from backend engine so I could not figure out why it happened. Could you guys help me out?
Thanks
P/s: I change to resnet50 and change the num_classes to 10 in the last fc layer.