conmmit

MickyasTA · Apr 27, 2024 · 00b5f8c · 00b5f8c
1 parent c0141ca
commit 00b5f8c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/__pycache__/Agent_Actor_Critic.cpython-311.pyc b/__pycache__/Agent_Actor_Critic.cpython-311.pyc
diff --git a/main.py b/main.py
@@ -8,14 +8,19 @@
 if __name__=='__main__':
     env=gym.make('CartPole-v1')
     agent=Agent(learning_rate=1e-5,gamma=0.99,n_actions=env.action_space.n)
-    n_games=1800
+    n_games=1400
     filename='cartpole.png'
     figure_file='plots/'+ filename
 
     best_score=env.reward_range[0]
     score_history=[]
     #load_checkpoint=False
     check_path_dir="tmp/actor_critic"
+
+    dir_name = 'D:\\MARS\\IRT_EXUPERY\\Actual Internship\\CODE\\Duckiebot_From-SIM-to-Real\\plots\\'
+    if not os.path.exists(dir_name):
+        os.makedirs(dir_name)
+
     if check_path_dir and os.path.exists(check_path_dir):
 
         agent.load_model()