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