start aerial for checking

Kaiyotech · Dec 21, 2022 · 43abd42 · 43abd42
1 parent 605d622
commit 43abd42
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 19 deletions.
diff --git a/Constants_aerial.py b/Constants_aerial.py
@@ -1,5 +1,5 @@
 FRAME_SKIP = 4
-TIME_HORIZON = 8  # horizon in seconds
+TIME_HORIZON = 6  # horizon in seconds
 T_STEP = FRAME_SKIP / 120   # real time per rollout step
 ZERO_SUM = True
 STEP_SIZE = 1_000_000

diff --git a/learner_aerial.py b/learner_aerial.py
@@ -51,10 +51,10 @@
         ent_coef=0.01,
     )
 
-    run_id = "aerial_run1"
+    run_id = "aerial_run1.01"
     wandb.login(key=os.environ["WANDB_KEY"])
     logger = wandb.init(dir="./wandb_store",
-                        name="Aerial_Run1",
+                        name="Aerial_Run1.01",
                         project="Opti",
                         entity="kaiyotech",
                         id=run_id,
@@ -131,7 +131,7 @@
         disable_gradient_logging=True,
     )
 
-    alg.load("aerial_saves/Opti_1665863109.179098/Opti_2300/checkpoint.pt")
+    alg.load("aerial_saves/Opti_1666209423.1454966/Opti_2380/checkpoint.pt")
     alg.agent.optimizer.param_groups[0]["lr"] = logger.config.actor_lr
     alg.agent.optimizer.param_groups[1]["lr"] = logger.config.critic_lr
 

diff --git a/worker_aerial.py b/worker_aerial.py
@@ -14,24 +14,24 @@
 from setter import CoyoteSetter
 import Constants_aerial
 import os
-set_num_threads(1)
 
+set_num_threads(1)
 
 if __name__ == "__main__":
     rew = ZeroSumReward(zero_sum=Constants_aerial.ZERO_SUM,
-                          goal_w=2,
-                          aerial_goal_w=5,
-                          double_tap_w=10,
-                          flip_reset_w=10,
-                          flip_reset_goal_w=20,
-                          punish_ceiling_pinch_w=0,
-                          concede_w=-10,
-                          velocity_bg_w=0.25,
-                          acel_ball_w=1,
-                          team_spirit=1,
-                          cons_air_touches_w=0.02,
-                          jump_touch_w=0.1,
-                          wall_touch_w=0.5,)
+                        goal_w=2,
+                        aerial_goal_w=5,
+                        double_tap_w=10,
+                        flip_reset_w=10,
+                        flip_reset_goal_w=20,
+                        punish_ceiling_pinch_w=0,
+                        concede_w=-10,
+                        velocity_bg_w=0.25,
+                        acel_ball_w=1,
+                        team_spirit=1,
+                        cons_air_touches_w=0.02,
+                        jump_touch_w=0.1,
+                        wall_touch_w=0.5, )
     frame_skip = Constants_aerial.FRAME_SKIP
     fps = 120 // frame_skip
     name = "Default"
@@ -62,7 +62,6 @@
             streamer_mode = True
             evaluation_prob = 0
             game_speed = 1
-            deterministic_streamer = True
             auto_minimize = False
 
     match = Match(