Create configs and perform basic setups.
(args)
| 378 | model = model.train().cuda() |
| 379 | |
| 380 | def setup(args): |
| 381 | """ |
| 382 | Create configs and perform basic setups. |
| 383 | """ |
| 384 | cfg = get_cfg() |
| 385 | cfg = LazyConfig.load(args.config_file) |
| 386 | cfg = LazyConfig.apply_overrides(cfg, args.opts) |
| 387 | # cfg.freeze() |
| 388 | default_setup(cfg, args) |
| 389 | setup_logger(output=cfg.OUTPUT_DIR, distributed_rank=comm.get_rank(), name="maskdino") |
| 390 | return cfg |
| 391 | |
| 392 | |
| 393 | def main(args=None): |