From 3b9fe874f11eeb450407aa83e785f6a6de485323 Mon Sep 17 00:00:00 2001
From: MaximeCerise <maxime.moreau.etu@gmail.com>
Date: Sun, 16 Mar 2025 18:24:30 +0100
Subject: [PATCH] save

---
 README.md                      | 4 +---
 evaluate_reinforce_cartpole.py | 2 +-
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 3c9e45e..773bba9 100644
--- a/README.md
+++ b/README.md
@@ -25,6 +25,4 @@ We finally have an evaluation with 100% of sucess:
 
 Here we set up a complete pipeline to solve Cartpole environment with A2C algorithm.
 
-Wandb has been set up to follow the learning phase. 
-https://wandb.ai/maximecerise-ecl/cartpole-a2c
-![alt text](saves/rollout.png)
+Wandb has been set up to track the learning phase : [WandB tacking](https://wandb.ai/maximecerise-ecl/cartpole-a2c?nw=nwusermaximecerise)
diff --git a/evaluate_reinforce_cartpole.py b/evaluate_reinforce_cartpole.py
index 42f4898..bf83fde 100644
--- a/evaluate_reinforce_cartpole.py
+++ b/evaluate_reinforce_cartpole.py
@@ -1,4 +1,4 @@
-import gym
+import gymnasium as gym
 import torch
 import numpy as np
 import torch
-- 
GitLab