Solved by Yang_TIAN in post #4 
              
              
                check the batch size of testdata loader？ 
testloader = torch.utils.data.DataLoader(dataset=testset, 
batch_size=4, 
shuffle=False) 
in case you set it in batch_size=1