Spaces:
Sleeping
Sleeping
Quentin Gallouédec
commited on
Commit
·
64d4c56
1
Parent(s):
158554b
fix atari record
Browse files- src/backend.py +2 -0
src/backend.py
CHANGED
@@ -461,6 +461,8 @@ def evaluate(
|
|
461 |
if episode_infos is not None:
|
462 |
logger.info(f"Atari Episode Score: {episode_infos['r']:.2f}")
|
463 |
logger.info("Atari Episode Length", episode_infos["l"])
|
|
|
|
|
464 |
|
465 |
if done and not is_atari and verbose > 0:
|
466 |
# NOTE: for env using VecNormalize, the mean reward
|
|
|
461 |
if episode_infos is not None:
|
462 |
logger.info(f"Atari Episode Score: {episode_infos['r']:.2f}")
|
463 |
logger.info("Atari Episode Length", episode_infos["l"])
|
464 |
+
episode_rewards.append(episode_infos["r"])
|
465 |
+
episode_lengths.append(episode_infos["l"])
|
466 |
|
467 |
if done and not is_atari and verbose > 0:
|
468 |
# NOTE: for env using VecNormalize, the mean reward
|