ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])
·
Error Note
- 전체 에러 문구 ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1]) 마지막 batch를 돌 때 데이터가 하나 밖에 남지 않아 발생하는 에러였습니다. - 해결 방법 데이터 갯수에 따라 마지막 batch에서 데이터가 하나만 남지 않도록 batch size를 적절히 조정해주어야 합니다.