torch.Size([512, 3, 256, 256]) torch.Size([512])

torch.Size([512, 3, 256, 256]) torch.Size([512])

torch.Size([512, 3, 256, 256]) torch.Size([512]) torch.Size([441, 3, 256, 256]) torch.Size([441])


Asserti: Batch size should be even when using this


아무래도 데이터셋 문제 같거든? 다른 데이터셋으로 학습할 때는 문제 없었는데 imagenet-mini로 바꾸니까 저러네

iteration 300번 반복이고 batch_size를 512로 줬는데

이미지 개수가 300 * 512 = 153600개보다 작으면 저런 오류 나는거 맞음?


난생 첨보는 에러라 당황스럽네