| 4 | |
| 5 | @HOOKS.register_module() |
| 6 | class TransferWeight(Hook): |
| 7 | |
| 8 | def __init__(self, every_n_inters=1): |
| 9 | self.every_n_inters=every_n_inters |
| 10 | |
| 11 | def after_train_iter(self, runner): |
| 12 | if self.every_n_inner_iters(runner, self.every_n_inters): |
| 13 | runner.eval_model.load_state_dict(runner.model.state_dict()) |
| 14 |
nothing calls this directly
no outgoing calls
no test coverage detected