()
| 114 | |
| 115 | |
| 116 | def parse_args(): |
| 117 | parser = argparse.ArgumentParser() |
| 118 | parser.add_argument("-c", "--config", help="yaml configuration file", required=True) |
| 119 | parser.add_argument("-s", "--seed", help="random seed for PyTorch", type=int, default=1024) |
| 120 | |
| 121 | args, unparsed = parser.parse_known_args() |
| 122 | # get dynamic arguments defined in the config file |
| 123 | vars = detect_variables(args.config) |
| 124 | parser = argparse.ArgumentParser() |
| 125 | for var in vars: |
| 126 | parser.add_argument("--%s" % var, default="null") |
| 127 | vars = parser.parse_known_args(unparsed)[0] |
| 128 | vars = {k: utils.literal_eval(v) for k, v in vars._get_kwargs()} |
| 129 | |
| 130 | return args, vars |
nothing calls this directly
no test coverage detected