[Deep Learning] [Pytorch] ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1
에러 발생 상황 : 1) Resnet 에 배치사이즈 100으로 이미지 데이터를 학습 시켰다. 2) 모델 학습이 완료된 후 학습한 모델에 1개의 테스트 인풋을 넣고 예측값을 확인하려했다. 3) 에러가 발생했다. Val
huidea.tistory.com
이걸 좀 오래 잡고 있었는데, model.eval() 을 통해서 해결하면 된다고 한다...
그래도 해결해서 다행이다.
'데이터 > Data Manipulation' 카테고리의 다른 글
| [einops] 차원 관리 툴 (0) | 2023.04.28 |
|---|---|
| [0] 데이터 품질 높이기 (0) | 2023.04.09 |
| Graph 자료 모음 (0) | 2022.10.20 |
| [Python] Missingno Package : Overview of new datasets (0) | 2022.10.20 |
| [Pandas] Stratified sampling with pd.DataFrame.sample() (0) | 2022.10.13 |
댓글