eval is executed after steps in config elapsed

2023-07-26 01:00:03 +09:00 · 2023-07-26 01:00:03 +09:00 · 6924abdd3e
commit 6924abdd3e
parent 43e1b2ab88
1 changed files with 2 additions and 1 deletions
--- a/dreamer.py
+++ b/dreamer.py
@ -323,7 +323,8 @@ def main(config):
        agent.load_state_dict(torch.load(logdir / "latest_model.pt"))
        agent._should_pretrain._once = False

-    while agent._step < config.steps:
+    # make sure eval will be executed once after config.steps
+    while agent._step < config.steps + config.eval_every:
        logger.write()
        print("Start evaluation.")
        eval_policy = functools.partial(agent, training=False)