-
질문 & 답변
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
라벨
21.08.21 01:16 작성 조회수 120
0
커스텀 폴더에 있는 이미지를 사용하는 과정입니다. 마지막 셀 2개를 보시면 trainset에 label이 0과1 인 이미지가 2개 들어가 있는 것을 알수 있습니다.
하지만 iter를 사용하여 돌려보면 tensor([0]) 이라고만 뜨고 나머지 라벨 1인 결과는 보여주지 않습니다. 여러번 실행 해봐도 0밖에 나오지않고 다음것을 보여주지 않는데 무슨 문제인가요?
답변을 작성해보세요.
0
딥러닝호형
지식공유자2021.08.21
무슨 오류인지 잘렸네요.
그런데 iter, next는 배치 크기만큼 불러오는게 맞습니다.
다만 현재 예시에서는 이미지 2장의 가로, 세로 크기가 달라서 여기서 문제가 생길 수도 있어요.
따라서 크기를 가로, 세로를 동일하게 하기위해 관련 오류가 나오게 되면 tr.Resize((16,16))로 바꿔주시면 됩니다.
0
ㄴ배치사이즈를 2로 늘려서 실행하니 이런 오류가 뜨네요 ㅠ0
답변 3