Li Peng 3e4da5f781

Replace deprecated pytorch methods (#1814) ...

* Replace deprecated pytorch methods

- torch.cuda.amp.GradScaler(...) => torch.amp.GradScaler("cuda", ...)
- torch.cuda.amp.autocast(...) => torch.amp.autocast("cuda", ...)

* Replace `with autocast(...)` with `with autocast("cuda", ...)`


Co-authored-by: Li Peng <lipeng@unisound.ai>

2024-12-16 10:24:16 +08:00

..

__init__.py

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +08:00

alignment.py

Strengthened style constraints (#1527)

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

asr_datamodule.py

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +08:00

beam_search.py

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +08:00

compute_ali.py

doc str fixes (#1241)

2023-09-07 16:34:53 +08:00

decode_gigaspeech.py

Strengthened style constraints (#1527)

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

decode.py

Fix typo (#1114)

2023-06-02 14:12:42 +08:00

decoder.py

Update decoder.py (#1262)

2023-09-20 08:15:54 +08:00

encoder_interface.py

Add Zipformer from Dan (#672)

2022-11-12 18:11:19 +08:00

export-onnx.py

Fix exporting decoder model to onnx (#1264)

2023-09-22 09:57:15 +08:00

export.py

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

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

finetune.py

Replace deprecated pytorch methods (#1814)

2024-12-16 10:24:16 +08:00

generate_model_from_checkpoint.py

Strengthened style constraints (#1527)

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

gigaspeech_scoring.py

Support fine-tuning (#944)

2023-03-17 13:44:29 +08:00

gigaspeech.py

Fix buffer size of DynamicBucketingSampler (#1468)

2024-01-21 02:10:42 +08:00

jit_pretrained.py

Use high_freq -400 in computing fbank features. (#1447)

2024-01-04 13:59:32 +08:00

joiner.py

Export streaming zipformer to ncnn (#906)

2023-02-13 23:41:43 +08:00

model.py

Replace deprecated pytorch methods (#1814)

2024-12-16 10:24:16 +08:00

my_profile.py

Strengthened style constraints (#1527)

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

onnx_check.py

Add ONNX support for Zipformer and ConvEmformer (#884)

2023-02-09 00:02:38 +08:00

onnx_decode.py

Strengthened style constraints (#1527)

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

onnx_pretrained.py

Add ONNX support for Zipformer and ConvEmformer (#884)

2023-02-09 00:02:38 +08:00

optim.py

fixes for init value of diagnostics.TensorDiagnosticOptions (#1269)

2023-09-24 17:06:47 +08:00

pretrained.py

Use high_freq -400 in computing fbank features. (#1447)

2024-01-04 13:59:32 +08:00

scaling_converter.py

Export streaming zipformer to ncnn (#906)

2023-02-13 23:41:43 +08:00

scaling.py

Replace deprecated pytorch methods (#1814)

2024-12-16 10:24:16 +08:00

test_compute_ali.py

Support batch-wise forced-alignment (#970)

2023-03-28 23:24:24 +08:00

test_model.py

Strengthened style constraints (#1527)

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

test_onnx.py

Fix CI tests (#1266)

2023-09-21 21:16:14 +08:00

train.py

Replace deprecated pytorch methods (#1814)

2024-12-16 10:24:16 +08:00

zipformer.py

Replace deprecated pytorch methods (#1814)

2024-12-16 10:24:16 +08:00