Why is the test data also added during model training? all_loader_train = itertools.chain(loader_train, loader_test)