Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/2615

Reviewed By: msaroufim

Differential Revision: D74413662

Pulled By: ZainRizvi

fbshipit-source-id: 4a0ae33d07604a41c9149983f631f1abbfd4540e