Function copy_optim_state_dict_cpu

PATH/core/utils.py:670–681

(state_dict)

    return new_state

def copy_optim_state_dict_cpu(state_dict):
    new_state = {}
    new_state['param_groups'] = copy.deepcopy(state_dict['param_groups'])
    new_state['state'] = {}
    for k,v in state_dict['state'].items():
        new_state['state'][k] = {}
        for name,x in v.items():
            if isinstance(x, torch.Tensor):
                new_state['state'][k][name] = x.cpu()
            else:
                new_state['state'][k][name] = copy.deepcopy(x)
    return new_state

def copy_optim_state_dict_cpu_fp16(state_dict):
    new_state = {}