From 7bc3e3177cf9dc5d53fbe9be667e51fa37729e96 Mon Sep 17 00:00:00 2001
From: Sofia Liguori <sofial@graphcore.ai>
Date: Fri, 13 Jan 2023 17:22:07 +0000
Subject: [PATCH] log wandb only if wandb is enabled

---
 nlp/gpt_j/popxl/run_finetuning_mnli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nlp/gpt_j/popxl/run_finetuning_mnli.py b/nlp/gpt_j/popxl/run_finetuning_mnli.py
index 0581fe164..e83371a47 100644
--- a/nlp/gpt_j/popxl/run_finetuning_mnli.py
+++ b/nlp/gpt_j/popxl/run_finetuning_mnli.py
@@ -149,7 +149,7 @@ def training(config: GPTJConfig, session: TaskSession, pretrained):
                               f"Duration: {duration:6.4f} s "
                               f"Throughput: {throughput:6.1f} samples/s ")
                 logging.info(result_str)
-                if popdist.getInstanceIndex() == 0:
+                if popdist.getInstanceIndex() == 0 and wandb.run is not None:
                     wandb.log(
                         {
                             "Loss": loss,