(cfg, dataset)
| 129 | |
| 130 | |
| 131 | def build_pretrain_solver(cfg, dataset): |
| 132 | if comm.get_rank() == 0: |
| 133 | logger.warning(dataset) |
| 134 | logger.warning("#dataset: %d" % (len(dataset))) |
| 135 | |
| 136 | task = core.Configurable.load_config_dict(cfg.task) |
| 137 | cfg.optimizer.params = task.parameters() |
| 138 | optimizer = core.Configurable.load_config_dict(cfg.optimizer) |
| 139 | solver = core.Engine(task, dataset, None, None, optimizer, **cfg.engine) |
| 140 | |
| 141 | return solver |
nothing calls this directly
no outgoing calls
no test coverage detected