diff --git a/egs/tal_csasr/ASR/lstm_transducer_stateless3/train.py b/egs/tal_csasr/ASR/lstm_transducer_stateless3/train.py index fa1b9d8e1..c6cdd7823 100755 --- a/egs/tal_csasr/ASR/lstm_transducer_stateless3/train.py +++ b/egs/tal_csasr/ASR/lstm_transducer_stateless3/train.py @@ -156,6 +156,13 @@ def add_model_arguments(parser: argparse.ArgumentParser): `grad_norm_threshold * median`, where `median` is the median value of gradient norms of all elememts in batch.""", ) + + parser.add_argument( + "--is-pnnx", + type=str2bool, + default=False, + help="Only used when exporting model with pnnx." + ) def get_parser():