Do you get the same error if you change that line to for i, (data, labels) in enumerate(full_train_loader): ? That should be at least part of the problem.  I can’t see the entire stack trace or code, so I don’t know if there are other issues as well.