From 190de6a213932080aad7e5dcefeaac762d9639a6 Mon Sep 17 00:00:00 2001
From: Ghelfi Manon <manon.ghelfi@ecl19.ec-lyon.fr>
Date: Wed, 8 Feb 2023 14:51:53 +0000
Subject: [PATCH] Update README.md
---
README.md | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/README.md b/README.md
index 4e79074..ddf5b86 100644
--- a/README.md
+++ b/README.md
@@ -36,7 +36,7 @@ push_to_hub(
After identifying with the command : `huggingface-cli login`
## Weights & Biases
-The run of the model is [here](https://wandb.ai/ghelfi/cartpole-training/runs/06exlpbm).
+The run of the model is [here](https://wandb.ai/ghelfi/cartpole-training/runs/t95xgk54).
Realized with the code below:
```
@@ -59,8 +59,7 @@ while True:
if done:
break
-print("Mean Reward: ", np.mean(rewards))
-wandb.log({'reward_mean': np.mean(rewards)})
+wandb.log({'rewards': np.sum(rewards)})
```
--
GitLab