4588ac6d2704f69dd9f527ae8145cd820c44667d,enjoy.py,,,#,13

Before Change

            torch.load(os.path.join(args.load_dir, args.env_name + ".pt"))


if len(env.observation_space.shape) == 1:
    env = VecNormalize(env, ret=False)
    env.ob_rms = ob_rms

    // An ugly hack to remove updates

After Change

render_func = None
tmp_env = env
while True:
    if hasattr(tmp_env, "envs"):
        render_func = tmp_env.envs[0].render
        break
    elif hasattr(tmp_env, "venv"):
        tmp_env = tmp_env.venv
    elif hasattr(tmp_env, "env"):
        tmp_env = tmp_env.env
    else:
        break

// We need to use the same statistics for normalization as used in training
actor_critic, ob_rms = \
            torch.load(os.path.join(args.load_dir, args.env_name + ".pt"))

if isinstance(env.venv, VecNormalize):

Instances