[해당 코드]

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

print(image_batch.size(), text_batch)

제가 baseline 코드를 돌렸을 때, 이 부분에서 막혀서 학습을 못 했었습니다.

여러 해결방법을 찾아 공유합니다.

• ipynb 코드 파일을 (train.csv, test.csv)파일이 있는 폴더에 넣기

• DataLoader의 num_workers를 0으로 설정

train_dataset = CustomDataset(train['img_path'].values, train['label'].values)

train_loader = DataLoader(train_dataset, batch_size = CFG['BATCH_SIZE'], shuffle=True, num_workers=0)

val_dataset = CustomDataset(val['img_path'].values, val['label'].values)

val_loader = DataLoader(val_dataset, batch_size = CFG['BATCH_SIZE'], shuffle=True, num_workers=0)

• next()에 iter(train_loader) 넣기

image_batch, text_batch = next(iter(train_loader))

• 이 방법은

                  AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next'

                  에러가 발생했을 때 사용하시면 됩니다.