luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

run abl_all

Browse Source
This commit is contained in:
luni
2023-02-24 20:31:04 +00:00
parent 0ce28a2e18
commit 784fe3eb45
5 changed files with 54 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/cvae_model.py
+2 -2
View File
@@ -179,10 +179,10 @@ class CVAENew(tf.keras.Model):
if config['use_curvature']:
self.curv_enc.trainable = False
self.encoder = VariationalEncoderNew(config['cvae']['encoder_depths'], config['cvae']['encoder_dims'], kernel_size=config['cvae']['kernel_size'])
self.encoder = VariationalEncoderNew(config['cvae']['encoder_depths'], config['cvae']['encoder_dims'], kernel_size=config['cvae']['kernel_size'], latent_dim=config['cvae']['latent_dim'])
self.decoder = DecoderNew(config['cvae']['decoder_depths'], config['cvae']['decoder_dims'], config['cvae']['kernel_size'])
self.reshape = tf.keras.layers.Reshape((8, 4, 64))
self.reshape = tf.keras.layers.Reshape((8, 4, config['cvae']['latent_dim']))
if config['avg_garment_path'] != '':
self.avg_garment = np.load(config['avg_garment_path'])