(cfg: Config)
| 20 | |
| 21 | @hydra.main(config_path="config", config_name="config", version_base=None) |
| 22 | def main(cfg: Config): |
| 23 | progress = ProgressLogger(cfg, exp_name=cfg.name) |
| 24 | device, use_ddp = get_device(cfg.device) |
| 25 | dataloader = create_dataloader(cfg.task.data, cfg.dataset, cfg.task.task, use_ddp) |
| 26 | model = create_model(cfg.model, class_num=cfg.dataset.class_num, weight_path=cfg.weight) |
| 27 | model = model.to(device) |
| 28 | |
| 29 | converter = create_converter(cfg.model.name, model, cfg.model.anchor, cfg.image_size, device) |
| 30 | |
| 31 | solver = ModelTrainer(cfg, model, converter, progress, device) |
| 32 | progress.start() |
| 33 | solver.solve(dataloader) |
| 34 | |
| 35 | |
| 36 | if __name__ == "__main__": |
no test coverage detected