@ptrblck
This was the line that throws this error.
  dataiter = iter(train_loader)
  images, labels = dataiter.next()

let me explain what I’m trying to do, and I would be happy if you have any suggestion.
I’m working with different color channels(RGB, HSV, LAB, YIQ …) and I have separated all the channels to see which ones are the most effective/powerful channels (Which in our case will give a higher accuracy). This peace of information will help me in my project!
So, what I need is to train only one channel, without duplication. Is that doable?