speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

decoding

Browse Source
This commit is contained in:
游雁 2024-06-13 17:45:33 +08:00
parent 7355e20503
commit d72df6cd2f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/train_utils/trainer_ds.py
View File

@ -30,8 +30,9 @@ def maybe_autocast(dtype=None, use_deepspeed=False):
yield
else:
if dtype == torch.float16 or dtype == torch.bfloat16:
with autocast(enabled=True, dtype=dtype):
yield
yield
# with autocast(enabled=True, dtype=dtype):
# yield
else:
yield