Fangjun Kuang fba5e67d5e

Fix CI tests. (#1974) ...

- Introduce unified AMP helpers (create_grad_scaler, torch_autocast) to handle 
  deprecations in PyTorch ≥2.3.0

- Replace direct uses of torch.cuda.amp.GradScaler and torch.cuda.amp.autocast 
  with the new utilities across all training and inference scripts

- Update all torch.load calls to include weights_only=False for compatibility with 
  newer PyTorch versions

2025-07-01 13:47:55 +08:00

..

__init__.py

Model average (#344)

2022-05-05 21:20:04 +08:00

asr_datamodule.py

Model average (#344)

2022-05-05 21:20:04 +08:00

beam_search.py

Model average (#344)

2022-05-05 21:20:04 +08:00

conformer.py

Model average (#344)

2022-05-05 21:20:04 +08:00

decode_stream.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

decode.py

Fix CI tests. (#1974)

2025-07-01 13:47:55 +08:00

decoder.py

Model average (#344)

2022-05-05 21:20:04 +08:00

encoder_interface.py

Model average (#344)

2022-05-05 21:20:04 +08:00

export.py

Use tokens.txt to replace bpe.model (#1162)

2023-08-12 16:53:59 +08:00

joiner.py

Model average (#344)

2022-05-05 21:20:04 +08:00

lstmp.py

fix export for stateless4 (#844)

2023-01-16 20:26:36 +08:00

model.py

Model average (#344)

2022-05-05 21:20:04 +08:00

my_profile.py

Strengthened style constraints (#1527)

2024-03-04 23:28:04 +08:00

optim.py

Model average (#344)

2022-05-05 21:20:04 +08:00

scaling_converter.py

fix export for stateless4 (#844)

2023-01-16 20:26:36 +08:00

scaling.py

Model average (#344)

2022-05-05 21:20:04 +08:00

streaming_beam_search.py

Add modified_beam_search for streaming decode (#489)

2022-07-25 16:53:23 +08:00

streaming_decode.py

streaming_decode.py, relax the audio range from [-1,+1] to [-10,+10] (#1448)

2024-01-05 10:21:27 +08:00

test_model.py

Using streaming conformer as transducer encoder (#380)

2022-06-28 00:18:54 +08:00

train.py

Fix CI tests. (#1974)

2025-07-01 13:47:55 +08:00