Quentin Gallouédec commited on
Commit
d7945e9
·
1 Parent(s): 64d4c56
Files changed (1) hide show
  1. src/backend.py +3 -3
src/backend.py CHANGED
@@ -460,7 +460,7 @@ def evaluate(
460
  episode_infos = infos[0].get("episode")
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
 
@@ -468,7 +468,7 @@ def evaluate(
468
  # NOTE: for env using VecNormalize, the mean reward
469
  # is a normalized reward when `--norm_reward` flag is passed
470
  logger.info(f"Episode Reward: {episode_reward:.2f}")
471
- logger.info("Episode Length", ep_len)
472
  episode_rewards.append(episode_reward)
473
  episode_lengths.append(ep_len)
474
  episode_reward = 0.0
@@ -477,7 +477,7 @@ def evaluate(
477
  # Reset also when the goal is achieved when using HER
478
  if done and infos[0].get("is_success") is not None:
479
  if verbose > 1:
480
- logger.info("Success?", infos[0].get("is_success", False))
481
 
482
  if infos[0].get("is_success") is not None:
483
  successes.append(infos[0].get("is_success", False))
 
460
  episode_infos = infos[0].get("episode")
461
  if episode_infos is not None:
462
  logger.info(f"Atari Episode Score: {episode_infos['r']:.2f}")
463
+ logger.info(f"Atari Episode Length {episode_infos["l"]}")
464
  episode_rewards.append(episode_infos["r"])
465
  episode_lengths.append(episode_infos["l"])
466
 
 
468
  # NOTE: for env using VecNormalize, the mean reward
469
  # is a normalized reward when `--norm_reward` flag is passed
470
  logger.info(f"Episode Reward: {episode_reward:.2f}")
471
+ logger.info(f"Episode Length {ep_len}")
472
  episode_rewards.append(episode_reward)
473
  episode_lengths.append(ep_len)
474
  episode_reward = 0.0
 
477
  # Reset also when the goal is achieved when using HER
478
  if done and infos[0].get("is_success") is not None:
479
  if verbose > 1:
480
+ logger.info(f"Success? {infos[0].get("is_success", False)}")
481
 
482
  if infos[0].get("is_success") is not None:
483
  successes.append(infos[0].get("is_success", False))