fix cvae

2023-01-26 14:45:35 +00:00
parent ea683ac030
commit 5d91b56a63
5 changed files with 34 additions and 15 deletions
@@ -182,7 +182,7 @@ class CVAENew(tf.keras.Model):
        self.encoder = VariationalEncoderNew(config['cvae']['encoder_dims'][-1])
        self.decoder = DecoderNew(config['cvae']['decoder_dims'][-1], is_cvae=True)

-        self.reshape = tf.keras.layers.Reshape((16, 8, config['cvae']['decoder_dims'][-1]))
+        self.reshape = tf.keras.layers.Reshape((8, 4, config['cvae']['decoder_dims'][-1]))

        if config['avg_garment_path'] != '':
            self.avg_garment = np.load(config['avg_garment_path'])