728x90
원인은 torch.view 에서 생기는 에러로 나의 경우는 .cuda()를 이용해 GPU 사용시 문제가 생겼다
해결방법
x.view(-1, ) -> x.contiguous().view(-1, )로 진행하면 해결된다
728x90
'AI | ML > 코드' 카테고리의 다른 글
| [PyTorch] RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation 에러 (0) | 2021.06.28 |
|---|---|
| [Keras] 'str' object has no attribute 'decode' (0) | 2021.06.22 |
| [Data Science] matplotplib을 이용한 데이터셋 가시화 (0) | 2021.02.17 |
| [Data Science] 이미지 데이터셋 내 평균, 표준편차 계산 (2) | 2021.01.04 |
| [TensorFlow] Model Save/Load 관련 (0) | 2020.08.27 |