Thank you for replying!!

Traceback (most recent call last):
File “/media/volume1/juna/.local/lib/python3.5/site-packages/torch/utils/data/_utils/worker.py”, line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File “/media/volume1/juna/.local/lib/python3.5/site-packages/torch/utils/data/_utils/worker.py”, line 99, in 
samples = collate_fn([dataset[i] for i in batch_indices])
File “…/datasets/MSDDataset.py”, line 264, in getitem
image = Image.fromarray(image)
File “/media/volume1/juna/.local/lib/python3.5/site-packages/PIL/Image.py”, line 2647, in fromarray
raise TypeError(“Cannot handle this data type”)
TypeError: Cannot handle this data type

Umm well, at the beginning I was trying to use “Image.fromarray(image)” and the error above is the error I got.
So I tried to use torchvision.transforms.ToPILImage instead but people on the internet said it also doesn’t work for float images.
Are there any other python packages I can use for float images?
Otherwise, I’m considering making custom tranform classes.