pytorch CNN err 질문드립니다!!

Question

6개 레이어의 모델을 학습시켜놓고 예측을 진행중입니다window에서 학습시켜 모델을 만들었고학습된 모델로 window에서 예측을 실행하였을 땐 잘 되었는데동일한 모델과 코드로 동일하게 linux에서 실행했을 때Given input zise: (512x1x6). Calculated out size:(512x0x3). Output size is too small에러가 발생했습니다. 5번째 레이어에서윈도우에서 돌릴 땐 512x1x6 -> 512x1x3으로 실행되는데linux에서는 512x1x6 -> 512x0x3으로 실행되는 것 같아요. 혹시 해결방법 아시는 분 있나요*재부팅 안되는 서버입니다.*겨우겨우 높은 정확도로 학습하게 만든 모델이라 모델을 변경하거나 다시 학습시키는 것을 제외한 방법이 있다면 알려주세요 ㅠㅠ

LLM만들자 · Accepted Answer

padding 값이 다른게 아닐까요? 원래 conv2d 적용되면 height와 width는 줄어들게 돼 있잖아요. 줄어들지 않게 하려면 padding이 필요하고요.

LLM만들자 · Answer

음? conv2d 인가요? 아님 conv1d 인가요...conv2d면 입력과 출력이 batch x channels x height x width 일텐데요... 버전 차이나 모델의 하이퍼 파라미터가 바뀐거 아닌가요?

ming9 · Answer

&#64;머신러닝conv2d 입니다. 입출력은 size는 앞에 batch를 생략하고 올렸습니다..헷갈리셨나면 죄송합니다 ! 하이퍼파라미터는 다 확인해봤는데 그대로인 것 같고 버전 차이...어떻게 해결할까요 ㅠㅠ

ming9 · Answer

&#64;머신러닝torch, cuda, torchvision 버전의 차이였어요감사합니다 머신러닝님!!

pytorch CNN err 질문드립니다!!

4개의 답변

#cnn 목록