Function load_model

ADHMR/lib/utils/function.py:208–214

(model,state)

        return model, model_cond, ema_helper, ema_helper_cond, optimizer_hyponet, optimizer_hrnet, start_epoch, step, loss, min_mpjpe_h36m, min_mpjpe_pw3d

def load_model(model,state):
    state_model = model.state_dict()
    for key in state_model.keys():
        if key in state.keys() and state_model[key].shape == state[key].shape:
            state_model[key] = state[key]
    model.load_state_dict(state_model)
    return model

def get_model_score(config, is_train = True, resume = False, resume_path = None):
    neighbour_matrix = get_neighbour_matrix_from_hand(parents,childrens,num_joints=config.hyponet.num_joints,num_edges=config.hyponet.num_twists,knn=config.scorenet.knn)