Add weights_only=True to torch.load #2741

Job	Run time
pytorch_tutorial_build_worker (1, 15, linux.g5.12xlarge.nvidia.gpu)	19m 51s
pytorch_tutorial_build_worker (2, 15, linux.g5.4xlarge.nvidia.gpu)	54m 20s
pytorch_tutorial_build_worker (3, 15, linux.g5.4xlarge.nvidia.gpu)	18m 48s
pytorch_tutorial_build_worker (4, 15, linux.g5.4xlarge.nvidia.gpu)	29m 58s
pytorch_tutorial_build_worker (5, 15, linux.g5.4xlarge.nvidia.gpu)	19m 28s
pytorch_tutorial_build_worker (6, 15, linux.g5.4xlarge.nvidia.gpu)	19m 14s
pytorch_tutorial_build_worker (7, 15, linux.4xlarge.nvidia.gpu)	19m 30s
pytorch_tutorial_build_worker (8, 15, linux.4xlarge.nvidia.gpu)	22m 57s
pytorch_tutorial_build_worker (9, 15, linux.4xlarge.nvidia.gpu)	33m 19s
pytorch_tutorial_build_worker (10, 15, linux.4xlarge.nvidia.gpu)	26m 2s
pytorch_tutorial_build_worker (11, 15, linux.4xlarge.nvidia.gpu)	42m 36s
pytorch_tutorial_build_worker (12, 15, linux.4xlarge.nvidia.gpu)	23m 46s
pytorch_tutorial_build_worker (13, 15, linux.4xlarge.nvidia.gpu)	20m 15s
pytorch_tutorial_build_worker (14, 15, linux.4xlarge.nvidia.gpu)	15m 24s
pytorch_tutorial_build_worker (15, 15, linux.4xlarge.nvidia.gpu)	23m 24s
pytorch_tutorial_build_manager	19m 0s
	6h 47m 52s

Provide feedback