wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-29 10:45:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Load evaluation configuration from checkpoint (#1392)

Browse Source
This commit is contained in:
Peter Schafhalter
2018-01-17 10:51:33 -08:00
committed by Eric Liang
parent b8811cbe34
commit 215d526e0d
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/eval.py
+10 -1
View File
@@ -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")