I made a minimum working example here, I use the code from (I copied the code to a jupyter notebook) https://github.com/pytorch/vision/blob/master/torchvision/models/quantization/resnet.py 1

However, it does not work, it raises this error:

And when removing the model.eval() code, it does not raises any error, but does not work as expected:

This is using the same ResNet, QuantizableResNet, and QuantizableBottleneck code used in the tutorials. These pictures were taken with this minimum reproducible code 3
Thoughts? @raghuramank100 @jerryzh168