Run the job given hyperparameters and utilities
(config, recorder, diagnostic_config)
| 559 | |
| 560 | |
| 561 | def run(config, recorder, diagnostic_config): |
| 562 | """Run the job given hyperparameters and utilities""" |
| 563 | with ( |
| 564 | diagnostic.diagnose(diagnostic_config), |
| 565 | maybe_record_goodput(recorder, GoodputEvent.JOB), |
| 566 | max_utils.maybe_get_transformer_engine_context(config), |
| 567 | maybe_monitor_goodput(config), |
| 568 | ): |
| 569 | train_loop(config, recorder) |
| 570 | |
| 571 | |
| 572 | def main(argv: Sequence[str]) -> None: |
no test coverage detected