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)