From eb521dfec0ae2661d0d5d14ab98684d3fbe07519 Mon Sep 17 00:00:00 2001
From: Majdi Karim <karim.majdi@etu.ec-lyon.fr>
Date: Tue, 5 Mar 2024 21:31:59 +0000
Subject: [PATCH] Update reinforce_cartpole.py

---
 reinforce_cartpole.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/reinforce_cartpole.py b/reinforce_cartpole.py
index 541eeef..a767524 100644
--- a/reinforce_cartpole.py
+++ b/reinforce_cartpole.py
@@ -1,3 +1,20 @@
+import gymnasium as gym
+import torch
+import torch.nn as nn
+import torch.optim as optim
+import torch.nn.functional as F
+from torch.distributions import Categorical
+import matplotlib.pyplot as plt
+
+
+
+
+
+
+
+
+
+
 # Create the environment
 env = gym.make("CartPole-v1", render_mode="human")
 
-- 
GitLab