From 215d526e0d605e2f090da3c7b1ec66c990bec89c Mon Sep 17 00:00:00 2001 From: Peter Schafhalter Date: Wed, 17 Jan 2018 10:51:33 -0800 Subject: [PATCH] Load evaluation configuration from checkpoint (#1392) --- python/ray/rllib/eval.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/python/ray/rllib/eval.py b/python/ray/rllib/eval.py index 9b34e63ff..ed5b6b1a4 100644 --- a/python/ray/rllib/eval.py +++ b/python/ray/rllib/eval.py @@ -7,6 +7,7 @@ from __future__ import print_function import argparse import gym import json +import os import ray from ray.rllib.agent import get_agent_class @@ -42,11 +43,19 @@ parser.add_argument( help="Run evaluation of the agent forever.") parser.add_argument( "--config", default="{}", type=json.loads, - help="Algorithm-specific configuration (e.g. env, hyperparams), ") + help="Algorithm-specific configuration (e.g. env, hyperparams). " + "Surpresses loading of configuration from checkpoint.") if __name__ == "__main__": args = parser.parse_args() + if not args.config: + # Load configuration from file + config_dir = os.path.dirname(args.checkpoint) + config_path = os.path.join(config_dir, "params.json") + with open(config_path) as f: + args.config = json.load(f) + if not args.env: if not args.config.get("env"): parser.error("the following arguments are required: --env")