(config, accelerator: Accelerator)
| 290 | |
| 291 | |
| 292 | def eval(config, accelerator: Accelerator): |
| 293 | _, _, _, _, validate_dataloader, _, _ = build_loader2(config) |
| 294 | model = build_model(config) |
| 295 | model, validate_dataloader = accelerator.prepare(model, validate_dataloader) |
| 296 | model = load_model_checkpoint(config, model, accelerator) |
| 297 | log_model_statistic(accelerator.unwrap_model(model)) |
| 298 | eval_epoch(config=config, data_loader=validate_dataloader, model=model, accelerator=accelerator) |
| 299 | |
| 300 | |
| 301 | def train(config, accelerator: Accelerator): |
no test coverage detected