It looks like your data variable in your for loop is your incrementer and the label variable in there contains the inputs and labels.  That’s what is being assigned in for (data, labels) in enumerate(full_train_loader) and I doubt that is what you want.