[tune/rllib] Add checkpoint eraser (#4490)

2026-08-10 12:30:14 +08:00 · 2019-04-06 20:01:54 -07:00
parent 7746d20d30
commit 820c71b7d0
6 changed files with 106 additions and 3 deletions
@@ -71,6 +71,8 @@ class Experiment(object):
                 sync_function=None,
                 checkpoint_freq=0,
                 checkpoint_at_end=False,
+                 keep_checkpoints_num=None,
+                 checkpoint_score_attr=None,
                 export_formats=None,
                 max_failures=3,
                 restore=None,
@@ -102,6 +104,8 @@ class Experiment(object):
            "sync_function": sync_function,
            "checkpoint_freq": checkpoint_freq,
            "checkpoint_at_end": checkpoint_at_end,
+            "keep_checkpoints_num": keep_checkpoints_num,
+            "checkpoint_score_attr": checkpoint_score_attr,
            "export_formats": export_formats or [],
            "max_failures": max_failures,
            "restore": restore