ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])

- 전체 에러 문구

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])

 

마지막 batch를 돌 때 데이터가 하나 밖에 남지 않아 발생하는 에러였습니다.

 

 

- 해결 방법

데이터 갯수에 따라 마지막 batch에서 데이터가 하나만 남지 않도록 batch size를 적절히 조정해주어야 합니다.