사물이미지 분류 코드를 학습하던 중 오류(?)가 있어서 문의 드립니다!

2024.05.16 01:05 778 조회

https://dacon.io/competitions/official/235874/codeshare/4584?page=1&dtype=recent

에서 진행하는 전이학습을 통해 사물이미지 분류 코드를 따라하며 공부하던 중 결과가 아래와 값이 NaN값이 나와서 뭐가 문제지..하고 살펴보니 무슨 multiprocessing이랑 JAX 호환 어쩌고 하는데 이것 때문일까요..?올바른 결과가 나오지 않는게..? 초보라 양해부탁드립니다ㅜ

로그인이 필요합니다
0 / 1000
DACON.YT
2024.05.16 16:56

안녕하세요. 
코드를 코랩에서 실행시켰을 때는 미지근한물 님과 같이 target 값이 NaN이 뜨지 않고, horse, cat 등 정상적으로 분류가 되는 것으로 확인됩니다.
NaN이 발생한 이유는 2가지 정도로 사료됩니다.
1. 코드 셀을 위에서부터 모든 코드를 순서대로 실행시키지 않음
2. 이미지를 불러올 때 경로를 잘못 설정 및 target 관련 정보를 잘못 불러옴

따라서, 링크로 주신 페이지의 코드(https://dacon.io/competitions/official/235874/codeshare/4584?page=1&dtype=recent)를 다운 받으신 후 코랩 혹은 개인 실행 환경에 그대로 업로드 하신 후 재시도 부탁드립니다.  

DACON.YT
2024.05.16 16:57

추가로, 저 경고메시지는 실제 학습을 할 때 문제가 발생하지 않으며 제가 실행할 때도 저 경고메시지가 떴으나 학습 및 추론은 정상적으로 됩니다.
만약 저 메시지가 문제가 된다고 판단하실 경우 파이토치의 `DataLoader()` 의 num_workers 파라미터를 0으로 설정해주세요.
감사합니다.

미지근한물
2024.05.16 20:21

친절한 답변 감사합니다!! 다시 해보겠습니다!