(
modify_parser: Optional[Callable[[argparse.ArgumentParser], None]] = None
)
| 435 | |
| 436 | |
| 437 | def cli_main( |
| 438 | modify_parser: Optional[Callable[[argparse.ArgumentParser], None]] = None |
| 439 | ) -> None: |
| 440 | parser = options.get_training_parser() |
| 441 | args = options.parse_args_and_arch(parser, modify_parser=modify_parser) |
| 442 | |
| 443 | cfg = convert_namespace_to_omegaconf(args) |
| 444 | |
| 445 | if args.profile: |
| 446 | with torch.cuda.profiler.profile(): |
| 447 | with torch.autograd.profiler.emit_nvtx(): |
| 448 | distributed_utils.call_main(cfg, main) |
| 449 | else: |
| 450 | distributed_utils.call_main(cfg, main) |
| 451 | |
| 452 | |
| 453 | if __name__ == "__main__": |
no test coverage detected