Error when load my dataset

[Haoyanlong]

I want to train the model by ImageNet dataset. I organize these training images path into ImageNet_train.flist，and I set the MODEL=1, MASK=1. I have the follow bug when load the dataset, but I test the size of inputed images as follows. Is it necessary to define the collate_fn function in the DataLoader? Could you help me? Thank you very much!

[knazeri]

@Haoyanlong It's not exactly clear to me where the bug is, but the last error message about dimension 1 of the tensor might suggest that some of your training set images might be grayscale and cannot be stacked with color images! I'm just speculating here, finding the exact error might need more investigation!
You do not need collate_fn function in DataLoader.

[Haoyanlong]

@knazeri OK，I see!Thank you!