Right, and then turn the batch_size back to normal for training? What is the usual way to do this? Are people just iterating over the batches first? I think knowing the initial loss is important.
edit: trying to assign a new batch size via batches_train.batch_size =   runs into error

ValueError: batch_size attribute should not be set after DataLoader is initialized