Bugfix - Models.dfl_h128

o2team · Aug 18, 2020 · 9c5568f · 9c5568f
1 parent f897562
commit 9c5568f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/train/model/dfl_h128.py b/plugins/train/model/dfl_h128.py
@@ -27,7 +27,7 @@ def encoder(self):
         var_x = Dense(8 * 8 * self.encoder_dim)(var_x)
         var_x = Reshape((8, 8, self.encoder_dim))(var_x)
         var_x = UpscaleBlock(self.encoder_dim)(var_x)
-        return KerasModel(input_, var_x)
+        return KerasModel(input_, var_x, name=self.name)
 
     def decoder(self, side):
         """ DFL H128 Decoder """