()
| 13 | |
| 14 | |
| 15 | def configure_arg_parser() -> ArgumentParser: |
| 16 | arg_parser = ArgumentParser() |
| 17 | arg_parser.add_argument("mode", help="Mode to run script", choices=["train", "test"]) |
| 18 | arg_parser.add_argument("-c", "--config", help="Path to YAML configuration file", type=str) |
| 19 | return arg_parser |
| 20 | |
| 21 | |
| 22 | def train_code2seq(config: DictConfig): |