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

fixed feature sizes, cvae working

Browse Source
This commit is contained in:
luni
2023-01-25 21:00:47 +00:00
parent cba9d8ce90
commit 46e37e22eb
12 changed files with 39 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/cond_encoder.py
+2 -2
View File
@@ -25,8 +25,8 @@ class AutoEncoderNew(tf.keras.Model):
def __init__(self, config):
super(AutoEncoderNew, self).__init__()
self.encoder : AEEncoderNew = AEEncoderNew(config['encoder_depths'], config['encoder_dims'], config['kernel_size'], config['activation'])
self.decoder : DecoderNew = DecoderNew(config['decoder_depths'], config['decoder_dims'], kernel_size=config['kernel_size'], is_label=config['is_label'])
self.encoder : AEEncoderNew = AEEncoderNew(config['bottleneck_size'], config['activation'])
self.decoder : DecoderNew = DecoderNew(config['bottleneck_size'], is_label=config['is_label'])
def call(self, inputs, training):