Quentin Gallouédec commited on
Commit
64d4c56
1 Parent(s): 158554b

fix atari record

Browse files
Files changed (1) hide show
  1. 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