from local

2025-12-11 06:55:27 +00:00 · 2023-05-24 20:59:02 +09:00 · 2023-05-24 20:59:02 +09:00 · 6fbbacf94c
commit 6fbbacf94c
parent 90c6e6531c
2 changed files with 5 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py
@ -1142,6 +1142,11 @@ def train_one_epoch(
                rank=rank,
            )
            del params.cur_batch_idx
+
+            if rank == 0: 
+                for i, lora in enumerate(lora_modules):
+                    lora.save_checkpoint(i, params.exp_dir)
+
            '''
            remove_checkpoints(
                out_dir=params.exp_dir,