Thanks it worked,
I am changing the resnet only in the last fc layer by making outputs = 10 as I am using it on CIFAR 10 set. But I end up getting `
RuntimeError: size mismatch, m1: [128 x 2048], m2: [512 x 1000] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:249`

error. Again this seems a version issue as I can run the code without error on my machine. What could be the reason??