Yes I saved them in text files. Initially I saved them as pickle objects in one file and loaded all in a list in the __init__() but that filled up my RAM and all crashed, so I then saved the images in separate text files.
I’m not sure how you want me to use torch.save for saving image files. I thought that was for saving the model? Sorry, please could you explain? Thanks.