MCPcopy Index your code
hub / github.com/hpcaitech/ColossalAI / eval

Method eval

colossalai/legacy/engine/_base_engine.py:215–218  ·  view source on GitHub ↗

Sets the model to evaluation mode.

(self)

Source from the content-addressed store, hash-verified

213 self._model.train()
214
215 def eval(self):
216 """Sets the model to evaluation mode."""
217 self.training = False
218 self._model.eval()

Callers 15

get_act_scalesMethod · 0.45
from_pretrainedMethod · 0.45
get_act_dictMethod · 0.45
initializeMethod · 0.45
_evalMethod · 0.45
predictMethod · 0.45
inferenceFunction · 0.45
new_from_pretrainedFunction · 0.45
init_modelMethod · 0.45
_init_modelMethod · 0.45
__init__Method · 0.45
generate_step_one_stageMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected