RuntimeError: stack expects each tensor to be equal size, but got [1, 256, 256] at entry 0 and [3, 256, 256] at entry 7

[gongmingbanzhi]

When I used my own dataset for training, an error occurred:

[caoyunkang]

It appears that your images are in grayscale rather than RGB format. You can update the dataset file to ensure images are read in RGB format.