Quentin Gallouédec commited on
Commit
8e630b3
1 Parent(s): 2a019f9

pong in eval

Browse files
Files changed (1) hide show
  1. src/evaluation.py +2 -1
src/evaluation.py CHANGED
@@ -21,6 +21,7 @@ ALL_ENV_IDS = [
21
  "Acrobot-v1",
22
  "Hopper-v4",
23
  "MsPacmanNoFrameskip-v4",
 
24
  ]
25
 
26
 
@@ -272,5 +273,5 @@ def evaluate(model_id, revision):
272
  episodic_returns.append(float(info["episode"]["r"]))
273
 
274
  results[env_id] = {"episodic_returns": episodic_returns}
275
- logger.info(f"Environment {env_id}: {np.mean(episodic_returns)} ± {np.mean(episodic_returns)}")
276
  return results
 
21
  "Acrobot-v1",
22
  "Hopper-v4",
23
  "MsPacmanNoFrameskip-v4",
24
+ "PongNoFrameskip-v4",
25
  ]
26
 
27
 
 
273
  episodic_returns.append(float(info["episode"]["r"]))
274
 
275
  results[env_id] = {"episodic_returns": episodic_returns}
276
+ logger.info(f"Environment {env_id}: {np.mean(episodic_returns)} ± {np.std(episodic_returns)}")
277
  return results