FunASR/funasr
BienBoy c1e365fea0
fix: resolve unexpected 'out of memory' issue in multi-GPU setup (#2373)
Fixed a bug where calling torch.cuda.empty_cache() caused extra memory usage on 'cuda:0', leading to unexpected 'out of memory' errors in multi-GPU environments.

Reference:
- https://github.com/pytorch/pytorch/issues/25752
- https://github.com/pytorch/pytorch/issues/144025
2025-02-01 23:29:34 +08:00
..
auto fix: resolve unexpected 'out of memory' issue in multi-GPU setup (#2373) 2025-02-01 23:29:34 +08:00
bin fix: resolve unexpected 'out of memory' issue in multi-GPU setup (#2373) 2025-02-01 23:29:34 +08:00
datasets Dev kws (#2105) 2024-09-25 15:10:50 +08:00
download emotion2vec 2024-12-17 11:15:53 +08:00
frontends feat: speed up fbank's lfr (#2246) 2024-11-30 13:05:39 +08:00
losses Dev gzf exp (#1654) 2024-04-24 16:03:38 +08:00
metrics fixbug 2024-06-24 10:20:05 +08:00
models fix: resolve unexpected 'out of memory' issue in multi-GPU setup (#2373) 2025-02-01 23:29:34 +08:00
optimizers Dev gzf exp (#1654) 2024-04-24 16:03:38 +08:00
schedulers Dev gzf exp (#1678) 2024-04-29 14:52:20 +08:00
tokenizer Dev kws (#2105) 2024-09-25 15:10:50 +08:00
train_utils step_or_epoch bugfix 2025-01-10 10:16:11 +08:00
utils 允许 model.generate 使用bytes io, 以便不写入文件,节省io时间 (#2343) 2024-12-29 22:33:22 +08:00
__init__.py Dev gzf deepspeed (#1833) 2024-06-20 17:09:33 +08:00
register.py v1.1.18 2024-12-12 11:37:23 +08:00
version.txt v1.2.3 2025-01-24 16:59:23 +08:00