Disable logging to avoid error with tensorboard

HazyResearch · Dec 16, 2020 · 1bb2542 · 1bb2542
1 parent af46c6d
commit 1bb2542
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pl_runner.py b/pl_runner.py
@@ -30,6 +30,7 @@ def pl_train(cfg, pl_model_class):
         limit_train_batches=cfg.train.limit_train_batches,
         track_grad_norm=2,
         **profiler_args,
+        logger=False,
     )
 
     trainer.fit(model)