Merge branch 'main' into f/vmap_optstate

ucl-dark · Oct 25, 2023 · 24ab517 · 24ab517
2 parents 409e7d6 + 00a6b6f
commit 24ab517
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pax/runners/runner_eval.py b/pax/runners/runner_eval.py
@@ -305,7 +305,10 @@ def run_loop(self, env, env_params, agents, num_episodes, watchers):
             self.model_path2 is not None and self.run_path2 is not None
         )
 
-        if watchers and not self.args.wandb.mode == "offline":
+        if watchers and self.args.wandb.mode not in [
+            "offline",
+            "disabled",
+        ]:
             wandb.restore(
                 name=self.model_path, run_path=self.run_path, root=os.getcwd()
             )