From 9c5568f887f8fadb7228a573174bf0752a7d0def Mon Sep 17 00:00:00 2001
From: torzdf <36920800+torzdf@users.noreply.github.com>
Date: Tue, 18 Aug 2020 22:26:29 +0100
Subject: [PATCH] Bugfix - Models.dfl_h128

---
 plugins/train/model/dfl_h128.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/plugins/train/model/dfl_h128.py b/plugins/train/model/dfl_h128.py
index 3afcf408c4..a11ef6d983 100644
--- 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 """