3f196bcdd1e0310208e8978cf4f1fb590614568b,utils.py,,load_model,#,15

Before Change

    return os.path.join(storage_dir(), "models", model_name+".pt")

def load_model(observation_space, action_space, from_path):
    if from_path == None or not(os.path.exists(from_path)):
        policy_net = Policy(observation_space, action_space)
        value_net = Value(observation_space)
    else:
        policy_net, value_net = pickle.load(open(from_path, "rb"))

    if ac_rl.use_gpu:
        policy_net = policy_net.cuda()
        value_net = value_net.cuda()

After Change

def load_model(observation_space, action_space, from_path):
    acmodel = ACModel(observation_space, action_space)
    if from_path != None and os.path.exists(from_path):
        acmodel.load_state_dict(torch.load(from_path))
    return acmodel

def save_model(acmodel, to_path):
    dirname = os.path.dirname(to_path)

Instances