Add arguments to benchmark to run pytext models. Output results in ms. (#30273)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30273
Pytext models expect input of the form `1xlength` and another input specifying the length.
Add the `pytext_len` argument to specify this.
ghstack-source-id: 94383501
Test Plan: ./speed_benchmark_torch --model model.pt --input_dims "1,4" --input_type int64 --warmup 10 --iter 10 --report_pep=true --pytext_len=4
Reviewed By: iseeyuan
Differential Revision: D18646028
fbshipit-source-id: 7d5fe0c36da6e5f7b0261619ce4784a46b70f3d8