3f196bcdd1e0310208e8978cf4f1fb590614568b,utils.py,,save_model,#,28

Before Change

    return policy_net, value_net

def save_model(policy_net, value_net, to_path):
    if ac_rl.use_gpu:
        policy_net.cpu(), value_net.cpu()
    
    dirname = os.path.dirname(to_path)
    if not(dirname):
        os.makedirs(dirname)

After Change

    dirname = os.path.dirname(to_path)
    if not(os.path.isdir(dirname)):
        os.makedirs(dirname)
    torch.save(acmodel.state_dict(), to_path)

Instances