Not seeing any performance improvement on resnet50 with torch.jit.script

jit
1

I tried to trace a modified resnet50 model ( taken from the torchray repo) . i havent observed any average improvements in forward pass upon doing this. my code is here: Google Colab

can anyone help with what i might be doing wrong?