From 89d677873eb1c99070bd9a33a36f4c6415396756 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Sun, 3 Jan 2021 20:17:04 +0800
Subject: Delete dead training judge

---
 models/auto_encoder.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/models/auto_encoder.py b/models/auto_encoder.py
index ac3cfdf..eaac2fe 100644
--- a/models/auto_encoder.py
+++ b/models/auto_encoder.py
@@ -122,12 +122,11 @@ class AutoEncoder(nn.Module):
         self.encoder = Encoder(channels, feature_channels, embedding_dims)
         self.decoder = Decoder(embedding_dims, feature_channels, channels)
 
-        if self.training:
-            f_c_dim = embedding_dims[1]
-            self.classifier = nn.Sequential(
-                nn.LeakyReLU(0.2, inplace=True),
-                BasicLinear(f_c_dim, num_class)
-            )
+        f_c_dim = embedding_dims[1]
+        self.classifier = nn.Sequential(
+            nn.LeakyReLU(0.2, inplace=True),
+            BasicLinear(f_c_dim, num_class)
+        )
 
     def forward(self, x_c1_t1, x_c1_t2, x_c2_t2, y=None):
         # x_c1_t2 is the frame for later module
-- 
cgit v1.2.3