베이스라인 돌리려하는데

2023 교원그룹 AI OCR 챌린지

베이스라인 돌리려하는데

구르릉

2022.12.28 15:14 3,380 Views

image_batch, text_batch = next(iter(train_loader))

print(image_batch.size(), text_batch)

여기서 너무 오래걸리고 안넘어가는데 원래그런가요..?

Comments 7개

로그인이 필요합니다

comment

0 / 1000

simsimhae

2022.12.28 17:11

https://www.inflearn.com/questions/40850/멀티프로세싱-에러-질문입니다 를 보고 참고했구요
저도 정확히는 몰라서

train_dataset = CustomDataset(train['img_path'].values, train['label'].values)
train_loader = DataLoader(train_dataset, batch_size = CFG['BATCH_SIZE'], shuffle=True)

val_dataset = CustomDataset(val['img_path'].values, val['label'].values)
val_loader = DataLoader(val_dataset, batch_size = CFG['BATCH_SIZE'], shuffle=True)
로
num_workers 를 없애서 강제로 바꾸니까 되기는 합니다 근본적인 해결법은 아닙니다.

simsimhae

2022.12.28 17:35

제 환경은 gpu 1개(3090) window입니다

구르릉

2022.12.28 17:14

감사합니다 ㅠㅠ 덕분에 해결됬습니다.

수리나칼루세

2022.12.29 19:01

삭제된 댓글입니다

DL_SM

2022.12.28 17:51

image_batch, text_batch = iter(train_loader).next()
print(image_batch.size(), text_batch)

이 방법도 시도해보세요!

교교교

2022.12.30 10:40

저도
image_batch, text_batch = iter(train_loader).next() 이 소스코드에서
아래 에러 나는데, 위 댓글 봐도 모르겠어요 도와주세요..

'_MultiProcessingDataLoaderIter' object has no attribute 'next'

ssung

2022.12.30 11:33

image_batch, text_batch = next(iter(train_loader))

로 변경해보세요. 또는 다음 링크 참조하시면 도움 될 듯 합니다.

https://dacon.io/competitions/official/236042/talkboard/407531?page=1&dtype=recent

📣 댓글 작성 창의 위치가 댓글 리스트 상단으로 이동하였습니다!

List

📣데이크루 4기 팀장에 지원 해주세요!(마감 2023년 1월 6일)

Competition - 데이크루 4기

현재 글

베이스라인 돌리려하는데

Competition - 2023 교원그룹 AI OCR 챌린지

Likes 12