Reset the model engine to the initial state. If the engine is not initialized, we initialize it. Otherwise, reload ckpt and reset states
(self)
| 116 | |
| 117 | @register(dispatch_mode=Dispatch.ONE_TO_ALL) |
| 118 | def reset(self): |
| 119 | """ |
| 120 | Reset the model engine to the initial state. If the engine is not initialized, |
| 121 | we initialize it. Otherwise, reload ckpt and reset states |
| 122 | """ |
| 123 | self.engine.initialize() |
| 124 | |
| 125 | def _postprocess_output(self, output, *, global_token_num, delta_time, forward_only): |
| 126 | """ |
no test coverage detected