Spaces:
Sleeping
Sleeping
Quentin Gallouédec
commited on
Commit
·
d7945e9
1
Parent(s):
64d4c56
fix log
Browse files- 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
|
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
|
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?
|
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))
|