4586366c1018257dd448d62f1fc7534e3f5a4338,enjoy.py,,,#,10

Before Change

                    help="environment to train on (default: PongNoFrameskip-v4)")
parser.add_argument("--load-dir", default="./trained_models/",
                    help="directory to save agent logs (default: ./trained_models/)")
parser.add_argument("--log-dir", default="/tmp/gym/",
                    help="directory to save agent logs (default: /tmp/gym)")
args = parser.parse_args()

try:
    os.makedirs(args.log_dir)

After Change

        else:
            return obs
    env._obfilt = types.MethodType(_obfilt, env)
    render_func = env.venv.envs[0].render
else:
    render_func = env.envs[0].render

Instances