torch.Size([32, 1024, 7, 7])
torch.Size([32, 1024, 7, 7])
torch.Size([32, 1024, 7, 7])
torch.Size([32, 1024, 7, 7])
torch.Size([32, 1024, 7, 7])
torch.Size([32, 1024, 7, 7])
torch.Size([20, 1024, 7, 7])
I printed out the shapes and I found that the last part of the batch size is different. how can I fix this problem? do I have to set batchsize that can be exactly divided by the dataset size? thanks for the advice?