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: latest
Versions
latest
stable
v0.10.3
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
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
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.