From 86332170fbf6f6cac4ef14c512a3594479effc20 Mon Sep 17 00:00:00 2001 From: Michael McKinsey Date: Thu, 11 Jun 2026 13:36:50 -0700 Subject: [PATCH] add train loss and val loss to epoch output --- ScaFFold/utils/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ScaFFold/utils/trainer.py b/ScaFFold/utils/trainer.py index a1f77f5..c5bc4cb 100644 --- a/ScaFFold/utils/trainer.py +++ b/ScaFFold/utils/trainer.py @@ -715,7 +715,7 @@ def train(self): # train_dice = float(train_dice_total.item() / len(self.train_loader)) self.log.info( - f" epoch {epoch} | train_dice_score {train_dice:.6f} | val_dice_score {val_score:.6f} | lr {self._current_learning_rate():.8f}" + f" epoch {epoch} | train_loss={overall_loss:.6f} | val_loss={val_loss_avg:.6f} | train_dice_score {train_dice:.6f} | val_dice_score {val_score:.6f} | lr {self._current_learning_rate():.8f}" ) self.log.debug(f" writing to csv at {self.outfile_path}") if self.world_rank == 0: