From 1bb254258e6fe300dbcc167d972c2dbcd1547928 Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Wed, 16 Dec 2020 14:40:06 -0800 Subject: [PATCH] Disable logging to avoid error with tensorboard --- pl_runner.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pl_runner.py b/pl_runner.py index 58898a1..2979a89 100644 --- 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)