decoding

游雁

2024-06-13 d72df6cd2f1914a749db0778233f3a66f447894a

decoding

 funasr/train_utils/trainer_ds.py

@@ -30,8 +30,9 @@
            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