Solved by ptrblck in post #2 
              
              
                I don’t know exactly what difference you would like to highlight. 
In the first approach train_inputs and train_labels seem to be undefined (as well as validation_x), so I assume you would like to use x_train etc.? 
Also, the DataLoader loop is different, since you are unpacking the values in the fi…