()
| 171 | |
| 172 | |
| 173 | def parse_args(): |
| 174 | parser = argparse.ArgumentParser( |
| 175 | description='Test if a given API model wrapper works properly') |
| 176 | parser.add_argument('config', help='Train config file path') |
| 177 | parser.add_argument('-n', '--non-interactive', action='store_true') |
| 178 | args = parser.parse_args() |
| 179 | return args |
| 180 | |
| 181 | |
| 182 | def parse_model_cfg(model_cfg: ConfigDict) -> Dict[str, ConfigDict]: |