인프런 커뮤니티 질문&답변

작성한 질문수

[파이토치] 실전 인공지능으로 이어지는 딥러닝 - 기초부터 논문 구현까지

데이터 불러오기(Data Loader)

라벨

작성

170

커스텀 폴더에 있는 이미지를 사용하는 과정입니다. 마지막 셀 2개를 보시면 trainset에 label이 0과1 인 이미지가 2개 들어가 있는 것을 알수 있습니다.

하지만 iter를 사용하여 돌려보면 tensor([0]) 이라고만 뜨고 나머지 라벨 1인 결과는 보여주지 않습니다. 여러번 실행 해봐도 0밖에 나오지않고 다음것을 보여주지 않는데 무슨 문제인가요?

답변 3

지식공유자

무슨 오류인지 잘렸네요.

그런데 iter, next는 배치 크기만큼 불러오는게 맞습니다.

다만 현재 예시에서는 이미지 2장의 가로, 세로 크기가 달라서 여기서 문제가 생길 수도 있어요.

따라서 크기를 가로, 세로를 동일하게 하기위해 관련 오류가 나오게 되면 tr.Resize((16,16))로 바꿔주시면 됩니다.

질문자

ㄴㄴㄴㄴ배치사이즈를 2로 늘려서 실행하니 이런 오류가 뜨네요 ㅠ

지식공유자

안녕하세요!

배치 사이즈를 1로 해서 1개만 나오는 겁니다. iter는 배치 개수만큼 데이터를 불러옵니다.

배치 사이즈를 2로 해보세요.

작성한 질문수