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.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
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.