Shortcuts

mmengine.runner.weights_to_cpu

mmengine.runner.weights_to_cpu(state_dict)[source]

Copy a model state_dict to cpu.

Parameters

state_dict (OrderedDict) – Model weights on GPU.

Returns

Model weights on GPU.

Return type

OrderedDict

Read the Docs v: v0.3.0
Versions
latest
stable
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.