diff --git a/egs/tedlium2/ASR/conformer_ctc3/.decode.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.decode.py.swp index 66b689c8c..6a3525dbe 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.decode.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.decode.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp index fffa5770b..a784df540 100644 Binary files a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp and b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp differ diff --git a/egs/tedlium2/ASR/conformer_ctc3/train.py b/egs/tedlium2/ASR/conformer_ctc3/train.py index 5815328e6..798022ed0 100755 --- a/egs/tedlium2/ASR/conformer_ctc3/train.py +++ b/egs/tedlium2/ASR/conformer_ctc3/train.py @@ -144,6 +144,25 @@ def add_model_arguments(parser: argparse.ArgumentParser) -> None: default=0, ) + parser.add_argument( + "--interctc", + type=str2bool, + default=False, + ) + + parser.add_argument( + "--condition", + type=str2bool, + default=False, + ) + + parser.add_argument( + "--interctc-weight", + type=float, + default=0.3, + ) + + def get_parser() -> argparse.ArgumentParser: parser = argparse.ArgumentParser( @@ -297,24 +316,6 @@ def get_parser() -> argparse.ArgumentParser: help="Whether to use half precision training.", ) - parser.add_argument( - "--interctc", - type=str2bool, - default=False, - ) - - parser.add_argument( - "--condition", - type=str2bool, - default=False, - ) - - parser.add_argument( - "--interctc-weight", - type=float, - default=0.3, - ) - add_model_arguments(parser) return parser