mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-08 01:22:22 +00:00
fba5e67d5e
- 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
25 lines
329 B
Plaintext
25 lines
329 B
Plaintext
kaldifst>1.7.0
|
|
kaldilm
|
|
kaldialign
|
|
num2words
|
|
kaldi-decoder
|
|
sentencepiece>=0.1.96
|
|
pypinyin==0.50.0
|
|
tensorboard
|
|
typeguard
|
|
dill
|
|
onnx>=1.15.0
|
|
onnxruntime>=1.16.3
|
|
onnxoptimizer
|
|
onnxsim
|
|
onnxconverter_common
|
|
|
|
# style check session:
|
|
black==22.3.0
|
|
isort==5.10.1
|
|
flake8==5.0.4
|
|
|
|
# cantonese word segment support
|
|
pycantonese==3.4.0
|
|
packaging
|