Hi,
If you are testing your model, you should set model.eval() mode. Then this would deactivate batchnorm and dropout layers. Then, you don’t need to have the same batch size between the test and train loader.