oh, you mean
in the train_loader , I use the optimizer to parameter dataloader.
in the test_loader , I don’t use any method to do parameter dataloader.
So it seems to be problematic?
but the ouputs can work normally?