Code
Hub
Workspaces
Connect
Indexed graphs
Engine
MCP
copy
hub
/
github.com/hpcaitech/ColossalAI
/ eval
Method
eval
tests/kit/model_zoo/custom/base.py:24–26 ·
view source on GitHub ↗
(self)
Source
from the content-addressed store, hash-verified
22
return
super().train(mode=mode)
23
24
def
eval(self):
25
self._use_checkpoint = False
26
return
super().eval()
Callers
15
test_torchvision_models
Function · 0.45
split_model_and_get_DAG
Function · 0.45
split_model_and_compare_output
Function · 0.45
split_model_and_compare_output
Function · 0.45
trace_and_compare
Function · 0.45
test_torchvision_models
Function · 0.45
trace_and_compare
Function · 0.45
trace_and_compare
Function · 0.45
trace_and_compare
Function · 0.45
trace_model_and_compare_output
Function · 0.45
trace_and_compare
Function · 0.45
assert_forward_equal
Function · 0.45
Calls
no outgoing calls
Tested by
15
test_torchvision_models
Function · 0.36
test_torchvision_models
Function · 0.36
trace_and_compare
Function · 0.36
trace_and_compare
Function · 0.36
trace_and_compare
Function · 0.36
trace_and_compare
Function · 0.36
assert_codegen_run
Function · 0.36
assert_codegen_run
Function · 0.36
assert_codegen_run
Function · 0.36
get_model
Function · 0.36
get_model
Function · 0.36
get_model
Function · 0.36