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