Your DataLoader will already provide batches so that you usually don’t need to unsqueeze the data.
It seems you have a batch of 4 color images. If that’s the case, just remove the unsqueeze lines of code and try to run your code again.