2022 관광데이터 AI 경진대회

베이스라인 코드 오류

2022.10.06 10:41 1,562 Views

베이스라인 코드 "run"부분에서 오류가 나는데

혹시 어떻게 해결해야 할까요?

로그인이 필요합니다
0 / 1000
전문가_전문가_Haaland
2022.10.14 13:58

train_dataset = CustomDataset(train_df['img_path'].values, train_vectors, train_df['cat3'].values, train_transform)
train_loader = DataLoader(train_dataset, batch_size = CFG['BATCH_SIZE'], shuffle=True, num_workers=6)

val_dataset = CustomDataset(val_df['img_path'].values, val_vectors, val_df['cat3'].values, test_transform)
val_loader = DataLoader(val_dataset, batch_size=CFG['BATCH_SIZE'], shuffle=False, num_workers=6) 

에서 num_workers =0 으로 바꿔보시길 추천드립니다