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

resnet 101 working

Browse Source
This commit is contained in:
luni
2023-02-21 22:30:32 +00:00
parent 8447b7c505
commit 9eb1183df4
4 changed files with 15 additions and 17 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['bottleneck_size'], config['activation'])
self.decoder : DecoderNew = DecoderNew(config['bottleneck_size'], is_label=config['is_label'])
self.encoder : AEEncoderNew = AEEncoderNew(config['layer_params'], config['bottleneck_size'], config['activation'])
self.decoder : DecoderNew = DecoderNew(config['layer_params'], config['bottleneck_size'], is_label=config['is_label'])
def call(self, inputs, training):