Fix bug #2458 (#2459)
Summary:
https://github.com/pytorch/benchmark/issues/2458
Pull Request resolved: https://github.com/pytorch/benchmark/pull/2459
Reviewed By: xuzhao9
Differential Revision: D63476542
Pulled By: kit1980
fbshipit-source-id: 01e9db9cb03d34e82a773897417df2ccda410634
