From 86c89c4f58dcb8f15a37773efd7455257a9be4d2 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@zalando.de>
Date: Mon, 15 Feb 2021 11:53:10 +0100
Subject: [PATCH] disable wandb by default

---
 pts/trainer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pts/trainer.py b/pts/trainer.py
index db77f34..c6e062e 100644
--- a/pts/trainer.py
+++ b/pts/trainer.py
@@ -23,6 +23,7 @@ class Trainer:
         learning_rate: float = 1e-3,
         weight_decay: float = 1e-6,
         maximum_learning_rate: float = 1e-2,
+        wandb_mode: str = "disabled",
         clip_gradient: Optional[float] = None,
         device: Optional[Union[torch.device, str]] = None,
         **kwargs,
@@ -35,7 +36,7 @@ class Trainer:
         self.maximum_learning_rate = maximum_learning_rate
         self.clip_gradient = clip_gradient
         self.device = device
-        wandb.init(**kwargs)
+        wandb.init(mode=wandb_mode, **kwargs)
 
     def __call__(
         self,