음성 감정 인식 AI 경진대회 월간 데이콘

알고리즘 | 음향 | 감정인식 | 분류 | Accuracy

Prize : 인증서
901명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

[Baseline] wav2vec2 모델 기반 Training Code

EISLab_이희원

2023.05.10 02:56 3,474 Views language

facebook에서 공개한 wav2vec2 모델을 활용하여 작성하였습니다.
- huggingface link : https://huggingface.co/facebook/wav2vec2-base

적은 리소스 환경에서도 큰 batch size 효과를 볼 수 있도록 gradient accumulation을 적용하였습니다.
'TOTAL_BATCH_SIZE'에 원하는 batch size를 선언하시고,
out of memory가 발생하지 않는 batch size를 'BATCH_SIZE'에 선언하시면 됩니다.
- gradient accumulation 설명 : https://towardsdatascience.com/what-is-gradient-accumulation-in-deep-learning-ec034122cfa

만약 추가로 궁금하신 점이 있다면, 댓글로 남겨주세요^^

Code