Revert "Fixes in Imagenet training script"#1225

msaroufim merged 1 commit intomainpytorch/examples:mainfrom

revert-1224-fix_gpu_countpytorch/examples:revert-1224-fix_gpu_countCopy head branch name to clipboard