fixing issue#2168 #3534

Job	Run time
pytorch_tutorial_build_worker (1, 15, linux.g5.12xlarge.nvidia.gpu)	11m 42s
pytorch_tutorial_build_worker (2, 15, linux.g5.4xlarge.nvidia.gpu)	50m 10s
pytorch_tutorial_build_worker (3, 15, linux.g5.4xlarge.nvidia.gpu)	19m 13s
pytorch_tutorial_build_worker (4, 15, linux.g5.4xlarge.nvidia.gpu)	29m 38s
pytorch_tutorial_build_worker (5, 15, linux.g5.4xlarge.nvidia.gpu)	19m 44s
pytorch_tutorial_build_worker (6, 15, linux.g5.4xlarge.nvidia.gpu)	17m 12s
pytorch_tutorial_build_worker (7, 15, linux.4xlarge.nvidia.gpu)	18m 23s
pytorch_tutorial_build_worker (8, 15, linux.4xlarge.nvidia.gpu)	12m 41s
pytorch_tutorial_build_worker (9, 15, linux.4xlarge.nvidia.gpu)	19m 21s
pytorch_tutorial_build_worker (10, 15, linux.4xlarge.nvidia.gpu)	14m 42s
pytorch_tutorial_build_worker (11, 15, linux.4xlarge.nvidia.gpu)	15m 4s
pytorch_tutorial_build_worker (12, 15, linux.4xlarge.nvidia.gpu)	29m 53s
pytorch_tutorial_build_worker (13, 15, linux.4xlarge.nvidia.gpu)	31m 3s
pytorch_tutorial_build_worker (14, 15, linux.4xlarge.nvidia.gpu)	43m 23s
pytorch_tutorial_build_worker (15, 15, linux.4xlarge.nvidia.gpu)	25m 6s
pytorch_tutorial_build_manager	19m 21s
	6h 16m 36s

Provide feedback