Fix the jit code for timm models. (#572)
Summary:
We didn't correctly set the jit script for the timm models. This PR makes sure all timm models are correctly jitted.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/572
Reviewed By: aaronenyeshi
Differential Revision: D32574781
Pulled By: xuzhao9
fbshipit-source-id: 4d731504e384dd740d98fec6b6ee75296d504946
