Changes to make the PyTorch tests run#13

RPrenger wants to merge 1 commit intofacebookresearch:mainfacebookresearch/dietgpu:mainfrom

RPrenger:fix_pytorch_testsRPrenger/dietgpu:fix_pytorch_testsCopy head branch name to clipboard