mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-05-24 21:08:50 +09:00
parent 6fbbacf94c
commit f3cbdad979
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp
View File

Binary file not shown.

2
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
View File

@ -1129,6 +1129,7 @@ def train_one_epoch(
and params.batch_idx_train % params.save_every_n == 0
):
params.cur_batch_idx = batch_idx
'''
save_checkpoint_with_global_batch_idx(
out_dir=params.exp_dir,
global_batch_idx=params.batch_idx_train,
@ -1141,6 +1142,7 @@ def train_one_epoch(
scaler=scaler,
rank=rank,
)
'''
del params.cur_batch_idx
if rank == 0: