diff --git a/examples/dqn/dqn_atari.py b/examples/dqn/dqn_atari.py index 3484d3f10ca..7d2850200f6 100644 --- 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