prep demo for restart

Kaiyotech · Jun 13, 2023 · 20ed564 · 20ed564
1 parent 6d1b416
commit 20ed564
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/learner_demo.py b/learner_demo.py
@@ -51,10 +51,10 @@
         ent_coef=0.01,
     )
 
-    run_id = "demo_run1.02"
+    run_id = "demo_run1.03"
     wandb.login(key=os.environ["WANDB_KEY"])
     logger = wandb.init(dir="./wandb_store",
-                        name="Demo_Run1.02",
+                        name="Demo_Run1.03",
                         project="Opti",
                         entity="kaiyotech",
                         id=run_id,
@@ -136,7 +136,7 @@
         disable_gradient_logging=True,
     )
 
-    alg.load("Demo_saves/Opti_1683728019.129012/Opti_2140/checkpoint.pt")
+    alg.load("Demo_saves/Opti_1683811323.9472697\Opti_2220/checkpoint.pt")
 
     alg.agent.optimizer.param_groups[0]["lr"] = logger.config.actor_lr
     alg.agent.optimizer.param_groups[1]["lr"] = logger.config.critic_lr