19f6bfac9d39794db025dc53f8921346f58080b9,enjoy.py,,,#,12

Before Change

        break
    elif hasattr(tmp_env, "venv"):
        tmp_env = tmp_env.venv
    elif hasattr(tmp_env, "env"):
        tmp_env = tmp_env.env
    else:
        break

After Change

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

vec_norm = get_vec_normalize(envs)
if vec_norm is not None:
    vec_norm.eval()
    vec_norm.ob_rms = ob_rms

recurrent_hidden_states = torch.zeros(1, actor_critic.recurrent_hidden_state_size)
masks = torch.zeros(1, 1)

if render_func is not None:

Instances