Step. Args: new_model: New Model instance.
(self, new_model)
| 126 | self.restore(ema_state_dict, build_fp32_params=False) |
| 127 | |
| 128 | def step(self, new_model): |
| 129 | """Step. |
| 130 | |
| 131 | Args: |
| 132 | new_model: New Model instance. |
| 133 | """ |
| 134 | self._step_internal(new_model) |
| 135 | |
| 136 | def reverse(self, model): |
| 137 | """ |
no test coverage detected