[Solved Error] RuntimeError: Expected 4-dimensional input for 4-dimensional weight 64 3 7 7, but got 3-dimensional input of size [3, 224, 224] instead
학습시킨 딥러닝 모델을 테스트 하려고 하니 아래와 같은 에러가 발생한다. RuntimeError: Expected 4-dimensional input for 4-dimensional weight 64 3 7 7, but got 3-dimensional input of size [3, 224, 224] instead 전처리 과정을 그대로 적용했는데 뭐가 문제일까 생각해보니 학습을 시킬때는 배치단위로 넘겼던 데이터를 테스트 단계에서는 1장 단위로 입력해서 그런 것 같다. 아래와 같이 차원을 늘려서 모델에 입력해주니 정상적으로 작동하였다. output = model(data[None, ...])