From 2b5bc2350df8ebb8e4e5722f5563f2e44823ac34 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Mon, 17 May 2021 17:19:20 +0800
Subject: Model modification

Reduce channels in the auto-encoder and add more layers
---
 models/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models/model.py')

diff --git a/models/model.py b/models/model.py
index 6118bdf..bf23428 100644
--- a/models/model.py
+++ b/models/model.py
@@ -366,7 +366,7 @@ class Model:
             for _f_p_c1_t2, _f_p_c2_t2 in zip(*f_loss[2])
         ]).sum()
 
-        return xrecon_loss, cano_cons_loss * 10, pose_sim_loss * 100
+        return xrecon_loss / 10, cano_cons_loss, pose_sim_loss * 10
 
     def _classification_loss(self, embedding, y):
         # Duplicate labels for each part
-- 
cgit v1.2.3