from local

2022-12-10 13:55:49 +09:00 · 2022-12-10 13:55:49 +09:00 · 088a522667
commit 088a522667
parent 4bf3376b1f
2 changed files with 1 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@ -887,6 +887,7 @@ def train_one_epoch(
    tb_writer: Optional[SummaryWriter] = None,
    world_size: int = 1,
    rank: int = 0,
+    wb = None,
 ) -> None:
    """Train the model for one epoch.