ShapeTextureDebiasedTraining
ShapeTextureDebiasedTraining copied to clipboard
Update PyTorch Compatibility and GPU Handling
Solving issues with the current version of PyTorch. Changes Made: Replaced transforms.RandomSizedCrop(224) with transforms.RandomResizedCrop(224) for random resizing and cropping of images. Deprecated the use of transforms.Scale and replaced it with transforms.Resize for image resizing. Set the CUDA device to 0 to avoid the error "No CUDA GPUs are available" in ImageNet.py scripts.