Wav2Vec2ProcessorWithLM

ingsol·2023년 4월 28일
0

ASR

목록 보기
12/18
post-custom-banner

여기 코드 따라하면됨.
https://huggingface.co/blog/wav2vec2-with-ngram#4-combine-an-n-gram-with-wav2vec2

  1. processor와 processor_lm을 따로 불러와 준다.

  1. without LM과 with LM결과 비교
    with LM은 logits의 argmax값인 predicted_ids가 아니라 logits을 바로 processor_lm에 넣어준다.

<참고>

gpu 개수 per_device_train_batch_size gradient_accumulation_steps = 160 정도 = 한 번에 돌아가는 batch_size

post-custom-banner

0개의 댓글