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