Spaces:
Runtime error
Runtime error
Quentin Gallouédec
commited on
Commit
·
8e630b3
1
Parent(s):
2a019f9
pong in eval
Browse files- 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.
|
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
|