From 0ec253e75ce28424c9e238f544f5aaac5e0ac4ef Mon Sep 17 00:00:00 2001 From: carefree0910 Date: Mon, 14 Oct 2024 16:35:41 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=87Constrained=20`TrainingLoopCallback?= =?UTF-8?q?`?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- core/learn/callbacks/defaults.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/learn/callbacks/defaults.py b/core/learn/callbacks/defaults.py index cd842aa..b2d0de1 100644 --- a/core/learn/callbacks/defaults.py +++ b/core/learn/callbacks/defaults.py @@ -116,7 +116,8 @@ def before_loop_with_loaders( return None loader = train_loader resumed_results = trainer.get_metrics(loader, trainer.config.valid_portion) - console.log("resumed metrics:", resumed_results) + if self.is_local_rank_0: + console.log("resumed metrics:", resumed_results) def before_gradient_update( self,