AttributeError: Can't get attribute

문제

졸업 프로젝트에서 노래와 광고를 분류하기 위해 텍스트 기반 분류기를 구현 중이다. 진행하다가 아래와 같은 오류를 만났다.

 

 

---

해결

찾아보니, 멀티 프로세싱..? 관련 문제라고 한다. 오류가 발생한 코드는 아래와 같은데

train_loader = DataLoader(train_data, batch_size=8, shuffle=True, num_workers=8)
test_loader = DataLoader(test_data, batch_size=8, shuffle=True, num_workers=8)

 

여기서 num_workers 부분에서 오류가 나는 것이었다.

 

아래와 같이 간단히 오류가 나는 부분을 '제거'하고 다시 실행해보니 문제 없이 작동했다.

train_loader = DataLoader(train_data, batch_size=8, shuffle=True, num_workers=0)
test_loader = DataLoader(test_data, batch_size=8, shuffle=True, num_workers=0)