Open ShellingFord221 opened 5 years ago
u can set batch_size =1 ,if you want test the data one by one
batch_size = 1 test_loader = torch.utils.data.DataLoader(dataset=test_dataset, batch_size=batch_size, # this line still remains shuffle=False)
Hi If I don't want to test the model in batch (i.e. test the test data one by one), should my code be:
And other codes don't change. Am I correct? Thanks!