save

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)

evaluate_reinforce_cartpole.py

-import gym
+import gymnasium as gym
 import torch
 import numpy as np
 import torch