Run fp16 resnet50 training in bench script (#17831)
Summary:
cc xw285cornell
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17831
Differential Revision: D14398532
Pulled By: bddppq
fbshipit-source-id: 37c03cc2eebe3a6083e05631cb6ff03474e4a8a2