Deeplabv3+ error

https://github.com/jfzhang95/pytorch-deeplab-xception

GitHub - jfzhang95/pytorch-deeplab-xception: DeepLab v3+ model in PyTorch. Support different backbones.

 

위 오픈소스를 사용해서 custom dataset 학습 시도중,

 

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

↑↑↑↑↑위와 같은 에러 발생

 

batch_size가 안맞아서 발생한다는데 이해할수없던 찰나에 퐆웅검색으로 찾아냄

 

https://github.com/msminhas93/DeepLabv3FineTuning/issues/5

Expected more than 1 value per channel when training · Issue #5 · msminhas93/DeepLabv3FineTuning

 

결론 : batch_size는 4이상으로 설정해라.