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