diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp index 3f4504aa5..86325ba07 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py index e1f1d1f42..12034bb33 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -138,6 +138,28 @@ def add_rep_arguments(parser: argparse.ArgumentParser): default=False, help="use sperate optimizer (enc / dec)", ) + parser.add_argument( + "--accum-grads", + type=int, + default=1, + help="accum-grad num.", + ) + parser.add_argument( + "--encoder-type", + type=str, + default='d2v', + help="Type of encoder (e.g. conformer, w2v, d2v...", + ) + parser.add_argument( + "--additional-block", + type=bool, + default=False, + ) + parser.add_argument( + "--freeze-finetune-updates", + type=int, + default=0 + )