MCPcopy Create free account
hub / github.com/InternLM/InternBootcamp / reset

Method reset

verl/verl/workers/engine_workers.py:118–123  ·  view source on GitHub ↗

Reset the model engine to the initial state. If the engine is not initialized, we initialize it. Otherwise, reload ckpt and reset states

(self)

Source from the content-addressed store, hash-verified

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 """

Callers 4

_build_engineMethod · 0.45
_init_engineMethod · 0.45
init_workersMethod · 0.45
init_modelMethod · 0.45

Calls 1

initializeMethod · 0.45

Tested by

no test coverage detected