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