From c225e119c20232f404397300333aea8fc96f4dbd Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 3 Mar 2021 16:41:35 +0100
Subject: [PATCH] Adjust metric tracker only for unsupervised domain
 adaptation.

---
 pysegcnn/core/trainer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index d361ed1..1a2ef97 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -1479,8 +1479,8 @@ class DomainAdaptationTrainer(ClassificationNetworkTrainer):
                         .format(repr(self.uda_loss_function)))
             self.uda_loss_function.device = self.device
 
-        # adjust metrics to accomodate domain adaptation metrics
-        self.tracker.train_metrics.extend(['cla_loss', 'uda_loss'])
+            # adjust metrics to accomodate domain adaptation metrics
+            self.tracker.train_metrics.extend(['cla_loss', 'uda_loss'])
 
     def _inp_uda(self, src_input, trg_input):
         """Domain adaptation at input feature level."""
-- 
GitLab