Returns: torch.nn.Module: It now calls :func:`detectron2.modeling.build_model`. Overwrite it if you'd like a different model.
(cls, cfg)
| 430 | |
| 431 | @classmethod |
| 432 | def build_model(cls, cfg): |
| 433 | """ |
| 434 | Returns: |
| 435 | torch.nn.Module: |
| 436 | |
| 437 | It now calls :func:`detectron2.modeling.build_model`. |
| 438 | Overwrite it if you'd like a different model. |
| 439 | """ |
| 440 | model = build_model(cfg) |
| 441 | logger = logging.getLogger(__name__) |
| 442 | logger.info("Model:\n{}".format(model)) |
| 443 | return model |
| 444 | |
| 445 | @classmethod |
| 446 | def build_optimizer(cls, cfg, model): |
no test coverage detected