Make timeout in resnet50_trainer configurable (#17058)
Summary:
xw285cornell petrex dagamayank
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17058
Differential Revision: D14068458
Pulled By: bddppq
fbshipit-source-id: 15df4007859067a22df4c6c407df4121e19aaf97
