From 37790b76d09f3aa6d1120e4a3dde2b855e600ca2 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Wed, 13 Oct 2021 11:56:32 +0000 Subject: [PATCH] argparse load conf from file --- deepspeech/training/cli.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/deepspeech/training/cli.py b/deepspeech/training/cli.py index e079293c..ca9652e3 100644 --- a/deepspeech/training/cli.py +++ b/deepspeech/training/cli.py @@ -28,6 +28,13 @@ class ExtendAction(argparse.Action): setattr(namespace, self.dest, items) +class LoadFromFile(argparse.Action): + def __call__(self, parser, namespace, values, option_string=None): + with values as f: + # parse arguments in the file and store them in the target namespace + parser.parse_args(f.read().split(), namespace) + + def default_argument_parser(): r"""A simple yet genral argument parser for experiments with parakeet. @@ -57,6 +64,8 @@ def default_argument_parser(): """ parser = argparse.ArgumentParser() parser.register('action', 'extend', ExtendAction) + parser.add_argument( + '--conf', type=open, action=LoadFromFile, help="config file.") train_group = parser.add_argument_group( title='Train Options', description=None)