adding tworr with PID and SAC environment, dev, openai#1

mahdinobar · Mar 22, 2024 · a44eafd · a44eafd
1 parent 1948d31
commit a44eafd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/spinup/examples/pytorch/inverted_pendulum_SAC.py b/spinup/examples/pytorch/inverted_pendulum_SAC.py
@@ -9,10 +9,10 @@
 from spinup.utils import plot
 import os
 
-TRAIN=False
-DEBUG=True
-env_fn = lambda: gym.make('Tworr-v1')
-exp_name = "Tworrv0_0"
+TRAIN=True
+DEBUG=False
+env_fn = lambda: gym.make('Tworr-v0')
+exp_name = "Tworrv0_1"
 if __name__ == '__main__':
     if TRAIN:
         # train