[간단해결] RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors

Willow·2022년 10월 3일
0

DEEP LEARNING

목록 보기
3/11

baseline model로 쓰기 위해 간단하게 audio classification 을 진행하는데 해당 에러가 나왔다. 평소에 쓰던 코드를 태스크만 바꾼 거라서 로우레벨 에러가 날 리 없다 생각했는데 num_labels를 잘못 설정해줘서 난 것이었다. 처음에 num_labels를 train_ds에 있는 레이블 개수로 설정했었는데, 설마 했는데 train_ds에 존재하는 레이블 개수와 test_ds에 존재하는 레이블 개수가 달랐다...

지금 실험하고 있는 speechocean762 데이터셋은 처음부터 test, train으로 분리되어 나와있는데 이렇게 train_ds와 test_ds의 레이블 개수가 달라도 되나 싶다.

profile
Speech Processing/AI/Linguistics/CS/etc.

0개의 댓글