From 900c0ebc8476f4479b233dec2e66380bb74607fd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carlos=20Mochol=C3=AD?= Date: Mon, 6 Mar 2023 19:50:21 +0100 Subject: [PATCH] Mention that the Trainer has started in barebones mode (#16926) Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> --- src/lightning/pytorch/trainer/trainer.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/lightning/pytorch/trainer/trainer.py b/src/lightning/pytorch/trainer/trainer.py index 430f0c6d8b..23a15eed40 100644 --- a/src/lightning/pytorch/trainer/trainer.py +++ b/src/lightning/pytorch/trainer/trainer.py @@ -844,6 +844,13 @@ class Trainer: self.fit_loop.min_epochs = min_epochs self.fit_loop.max_epochs = max_epochs + if self.barebones: + # no progress bar in barebones can make it look like the Trainer hung + rank_zero_info( + "`Trainer(barebones=True)` started running. The progress bar is disabled so you might want to" + " manually print the progress in your model." + ) + # clean hparams if hasattr(model, "hparams"): parsing.clean_namespace(model.hparams)