.

marrlab · Sep 17, 2024 · 88ec169 · 88ec169
1 parent c078bc4
commit 88ec169
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/domainlab/algos/trainers/train_coral.py b/domainlab/algos/trainers/train_coral.py
@@ -25,7 +25,8 @@ def cross_domain_mmd(self, tuple_data_domains_batch):
             for ind_domain_b in range(ind_domain_a, num_domains):
                 data_b, *_ = tuple_data_domains_batch[ind_domain_b]
                 feat_b = self.model.extract_semantic_feat(data_b)
-                list_cross_domain_mmd.append(sum(self.mmd(feat_a, feat_b)))
+                mmd = self.mmd(feat_a, feat_b)
+                list_cross_domain_mmd.append(sum(mmd))
         return list_domain_erm_loss, list_cross_domain_mmd
 
     def tr_epoch(self, epoch):