aerial setter changes 4860

Kaiyotech · Feb 17, 2023 · 60d54df · 60d54df
1 parent 168a3ca
commit 60d54df
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/learner_aerial.py b/learner_aerial.py
@@ -137,10 +137,10 @@
         disable_gradient_logging=True,
     )
 
-    alg.load("aerial_saves/Opti_1676493790.2287304/Opti_4370/checkpoint.pt")
+    alg.load("aerial_saves/Opti_1676558987.6488075/Opti_4860/checkpoint.pt")
     alg.agent.optimizer.param_groups[0]["lr"] = logger.config.actor_lr
     alg.agent.optimizer.param_groups[1]["lr"] = logger.config.critic_lr
 
-    alg.freeze_policy(30)
+    # alg.freeze_policy(30)
 
     alg.run(iterations_per_save=logger.config.save_every, save_dir="aerial_saves")
diff --git a/setter.py b/setter.py
@@ -6,7 +6,7 @@
 from rlgym_tools.extra_state_setters.augment_setter import AugmentSetter
 from rlgym.utils.state_setters.random_state import RandomState
 from mybots_statesets import GroundAirDribble, WallDribble, HalfFlip, Walldash, Wavedash, \
-    Curvedash, Chaindash, RandomEvenRecovery, RecoverySetter
+    Curvedash, Chaindash, RandomEvenRecovery, RecoverySetter, LixSetter
 
 
 class CoyoteSetter(DynamicGMSetter):
@@ -147,8 +147,9 @@ def __init__(self, mode, end_object_choice=None):
                             AugmentSetter(WallDribble(), True, False, False),
                             AugmentSetter(RandomState(cars_on_ground=False), True, False, False),
                             AugmentSetter(ReplaySetter(double_tap_replays[i], random_boost=True, remove_defender_weight=0.1), True, False, False),
+                            AugmentSetter(GroundAirDribble(), True, False, False),
                         ),
-                        (0.05, 0.30, 0.20, 0.20, 0.10, 0.01, 0.14)
+                        (0.05, 0.20, 0.14, 0.15, 0.15, 0, 0.21, 0.1)
                     )
                 )
         elif mode == "pinch":