[tune] Clean up result logging: move out of /tmp, add timestamp (#1297)

This commit is contained in:
Eric Liang
2017-12-15 14:19:08 -08:00
committed by GitHub
parent 12fdb3f53a
commit fbf1806b8a
11 changed files with 64 additions and 26 deletions
+15 -2
View File
@@ -12,6 +12,7 @@ from ray.rllib import _register_all
from ray.tune import Trainable, TuneError
from ray.tune import register_env, register_trainable, run_experiments
from ray.tune.registry import _default_registry, TRAINABLE_CLASS
from ray.tune.result import DEFAULT_RESULTS_DIR
from ray.tune.trial import Trial, Resources
from ray.tune.trial_runner import TrialRunner
from ray.tune.variant_generator import generate_trials, grid_search, \
@@ -63,6 +64,17 @@ class TrainableFunctionApiTest(unittest.TestCase):
"config": {"a": "b"},
}})
def testLogdir(self):
def train(config, reporter):
assert "/tmp/logdir/foo" in os.getcwd(), os.getcwd()
reporter(timesteps_total=1)
register_trainable("f1", train)
run_experiments({"foo": {
"run": "f1",
"local_dir": "/tmp/logdir",
"config": {"a": "b"},
}})
def testBadParams(self):
def f():
run_experiments({"foo": {}})
@@ -191,7 +203,9 @@ class VariantGeneratorTest(unittest.TestCase):
self.assertEqual(trials[0].config, {"foo": "bar", "env": "Pong-v0"})
self.assertEqual(trials[0].trainable_name, "PPO")
self.assertEqual(trials[0].experiment_tag, "0")
self.assertEqual(trials[0].local_dir, "/tmp/ray/tune-pong")
self.assertEqual(
trials[0].local_dir,
os.path.join(DEFAULT_RESULTS_DIR, "tune-pong"))
self.assertEqual(trials[1].experiment_tag, "1")
def testEval(self):
@@ -207,7 +221,6 @@ class VariantGeneratorTest(unittest.TestCase):
self.assertEqual(len(trials), 1)
self.assertEqual(trials[0].config, {"foo": 4})
self.assertEqual(trials[0].experiment_tag, "0_foo=4")
self.assertEqual(trials[0].local_dir, "/tmp/ray/")
def testGridSearch(self):
trials = generate_trials({