Add mode to get_best_logdir

huawei-noah · Jan 19, 2021 · 3475527 · 3475527
1 parent 8d564be
commit 3475527
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/rllib.py b/examples/rllib.py
@@ -139,7 +139,7 @@ def main(
 
     print(analysis.dataframe().head())
 
-    best_logdir = Path(analysis.get_best_logdir("episode_reward_max"))
+    best_logdir = Path(analysis.get_best_logdir("episode_reward_max", mode="max"))
     model_path = best_logdir / "model"
 
     copy_tree(str(model_path), save_model_path, overwrite=True)