Shortcuts

mmengine.runner.weights_to_cpu

mmengine.runner.weights_to_cpu(state_dict)[源代码]

Copy a model state_dict to cpu.

参数

state_dict (OrderedDict) – Model weights on GPU.

返回

Model weights on GPU.

返回类型

OrderedDict

Read the Docs v: v0.7.0
Versions
latest
stable
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.