Set model name early to keep warmup and main model same (#159231)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/159231
Approved by: https://github.com/williamwen42
ghstack dependencies: #159209
Reviewed By: ZainRizvi
Differential Revision: D79112571
fbshipit-source-id: fbaf66b7e28f639684fe4e59f6c54b8a69736b6a