I am getting the same when I change the format. I think that there is something wrong in the class ‘imagedataset2’ because the error is in the line
----> 4 for batch in trainLoader:
but I can’t spot that out. I am actually a beginner in pytorch. Please let me know your thoughts here.