fix end-of-life

pytorch · Nov 30, 2023 · 75625bd · 75625bd
1 parent 06cc740
commit 75625bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/dqn/dqn_atari.py b/examples/dqn/dqn_atari.py
@@ -86,7 +86,7 @@ def main(cfg: "DictConfig"):  # noqa: F821
         loss_function="l2",
         delay_value=True,
     )
-    loss_module.set_keys(done="eol", terminated="eol")
+    loss_module.set_keys(done="end-of-life", terminated="end-of-life")
     loss_module.make_value_estimator(gamma=cfg.loss.gamma)
     target_net_updater = HardUpdate(
         loss_module, value_network_update_interval=cfg.loss.hard_update_freq