We were using old version of torchvision.transform which only support PIL image.
https://pytorch.org/docs/1.6.0/torchvision/transforms.html#transforms-on-torch-tensor
either change to PIL image when loading the image or do our own transformation (which is probably bad idea).
Need to see if we can only upgrade torchvision only.
If try to transformToPIL, getting this error https://discuss.pytorch.org/t/error-while-using-transforms-topilimage-and-randomresizedcrop/12861
Colorado's change with custom data loader:
Berkeley-Data/OpenSelfSup#3
We were using old version of torchvision.transform which only support PIL image.
https://pytorch.org/docs/1.6.0/torchvision/transforms.html#transforms-on-torch-tensor
either change to PIL image when loading the image or do our own transformation (which is probably bad idea).
Need to see if we can only upgrade torchvision only.
If try to transformToPIL, getting this error https://discuss.pytorch.org/t/error-while-using-transforms-topilimage-and-randomresizedcrop/12861
Colorado's change with custom data loader:
Berkeley-Data/OpenSelfSup#3