ShapeTextureDebiasedTraining icon indicating copy to clipboard operation
ShapeTextureDebiasedTraining copied to clipboard

Update PyTorch Compatibility and GPU Handling

Open f-amerehi opened this issue 1 year ago • 0 comments

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.

f-amerehi avatar Jul 19 '23 09:07 f-amerehi