from local

2025-12-11 06:55:27 +00:00 · 2023-04-11 15:36:55 +09:00 · 2023-04-11 15:36:55 +09:00 · 8527f5d8c9
commit 8527f5d8c9
parent 9fe251dee1
2 changed files with 1 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
@ -993,6 +993,7 @@ def train_one_epoch(
    world_size: int = 1,
    rank: int = 0,
    wb = None,
+    prompt = None,
 ) -> None:
    """Train the model for one epoch.