From 89d677873eb1c99070bd9a33a36f4c6415396756 Mon Sep 17 00:00:00 2001 From: Jordan Gong 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(-) (limited to 'models') 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