From 394a6c551b22dc30bdd4aa75b085236bb63c7ed5 Mon Sep 17 00:00:00 2001 From: SarahAlidoost Date: Wed, 19 Aug 2026 11:33:39 +0200 Subject: [PATCH] fix writer in train module --- climanet/train.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/climanet/train.py b/climanet/train.py index 0dd2913..682133f 100644 --- a/climanet/train.py +++ b/climanet/train.py @@ -154,6 +154,9 @@ def train_monthly_model( avg_train_loss = epoch_loss.item() / num_batches avg_epoch_loss = avg_train_loss # Initially use training loss + if training_config.store_logs: + writer.add_scalar("Loss/train", avg_train_loss, epoch) + # Validation loss (optional) if dataset_validation is not None: prediction_config = PredictionConfig( @@ -175,6 +178,9 @@ def train_monthly_model( ) avg_epoch_loss = avg_val_loss + if training_config.store_logs: + writer.add_scalar("Loss/validation", avg_val_loss, epoch) + if ( training_config.verbose and epoch % training_config.verbose_epoch_interval == 0