openspeech-team / openspeech

Open-Source Toolkit for End-to-End Speech Recognition leveraging PyTorch-Lightning and Hydra.
https://openspeech-team.github.io/openspeech/
MIT License
670 stars 112 forks source link

mixed precision 학습 관련 #224

Open laetokang opened 6 months ago

laetokang commented 6 months ago

❓ Questions & Help

Transducer (RNN-T, Conformer-T) 계열 모델로 학습하려고 하는데 학습 시간이 오래 걸려서 mixed precision 사용하려고 합니다.

계속 아래와 같은 에러가 발생하는데, Transducer 모델은 이 방법을 사용할 수 없는걸까요?

RuntimeError: Input and parameter tensors are not the same dtype, found input tensor with Float and parameter tensor with Half

Details