zr_jin
f1e9a7521a
Merge f3b918452ac6b210c25651f974bc8e8141aff101 into fba5e67d5e14c808cea7f2bf5ccc7fa0c248cc5c
2025-07-05 02:50:54 +08:00
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
Fangjun Kuang
d4d4f281ec
Revert "Replace deprecated pytorch methods ( #1814 )" ( #1841 )
...
This reverts commit 3e4da5f78160d3dba3bdf97968bd7ceb8c11631f.
2024-12-18 16:49:57 +08:00
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
zr_jin
a394bf7474
fixed gss scripts for alimeeting and ami recipes ( #1749 )
2024-09-08 20:35:07 +08:00
zr_jin
eb132da00d
additional instruction for the grad_scale is too small error ( #1550 )
2024-03-14 11:33:49 +08:00
Yifan Yang
5dfc3ed7f9
Fix buffer size of DynamicBucketingSampler ( #1468 )
...
* Fix buffer size
* Fix for flake8
---------
Co-authored-by: yifanyeung <yifanyeung@yifanyeung.local>
2024-01-21 02:10:42 +08:00
jinzr
f3b918452a
arch align
2023-10-19 11:09:20 +08:00
jinzr
f3f0dfc52d
minor updates
2023-10-18 15:07:12 +08:00
zr_jin
5340633654
Merge branch 'k2-fsa:master' into fix/ami
2023-10-18 02:01:04 +08:00
jinzr
014a59b597
Update prepare_ami_gss.sh
2023-10-17 21:58:48 +08:00
zr_jin
d2bd0933b1
Compatibility with the latest Lhotse ( #1314 )
2023-10-17 21:22:32 +08:00
jinzr
e20af6f28b
Update decode.py
2023-10-15 22:56:06 +08:00
jinzr
71e8d2d8e2
init commit
2023-10-15 16:04:00 +08:00
zr_jin
ef658d691e
fixes for init value of diagnostics.TensorDiagnosticOptions ( #1269 )
...
* fixes for `diagnostics`
Replace `2 ** 22` with `512` as the default value of `diagnostics.TensorDiagnosticOptions`
also black formatted some scripts
* fixed formatting issues
2023-09-24 17:06:47 +08:00
Fangjun Kuang
7b0afbdc16
Remove cur_batch_idx ( #1102 )
2023-05-30 14:49:54 +08:00
Fangjun Kuang
f5de2e90c6
Fix style issues. ( #937 )
2023-03-08 22:56:04 +08:00
pehonnet
07243d136a
remove key from result filename ( #936 )
...
Co-authored-by: pe-honnet <pe.honnet@telepathy.ai>
2023-03-08 21:06:07 +08:00
armusc
d65fe17d27
Update train.py with parameters_names as required by optimizer initialization ( #742 )
...
* Update train.py
2022-12-08 20:21:51 +08:00
Desh Raj
db75627e92
[recipe] AMI Zipformer transducer ( #698 )
...
* remove unnecessary changes
* add AMI prepare scripts
* add zipformer scripts for AMI
* added logs and pretrained model
* minor fix
* remove unwanted changes
* fix missing link
* make suggested changes
* update results
2022-11-26 10:00:45 +08:00
