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

Before Change

env = make_env(args.env_name, args.seed, 0, None, args.add_timestep)
env = DummyVecEnv([env])

actor_critic, ob_rms = \
            torch.load(os.path.join(args.load_dir, args.env_name + ".pt"))

After Change

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

Instances