hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

eval is executed after steps in config elapsed

Browse Source
This commit is contained in:
NM512 2023-07-26 01:00:03 +09:00
parent 43e1b2ab88
commit 6924abdd3e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dreamer.py
View File

@ -323,7 +323,8 @@ def main(config):
agent.load_state_dict(torch.load(logdir / "latest_model.pt")) agent.load_state_dict(torch.load(logdir / "latest_model.pt"))
agent._should_pretrain._once = False 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() logger.write()
print("Start evaluation.") print("Start evaluation.")
eval_policy = functools.partial(agent, training=False) eval_policy = functools.partial(agent, training=False)