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

trying to get resnet working

Browse Source
This commit is contained in:
luni
2023-02-22 10:33:29 +00:00
parent e0a5c9a2e3
commit bb7128f589
4 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/base.py
+4 -4
View File
@@ -116,7 +116,7 @@ class VariationalEncoderNew(tf.keras.Model):
def __init__(self, features):
super(VariationalEncoderNew, self).__init__()
self.encoder = resnet_101(features)
self.encoder = resnet_18(features)
self.mean_branch = tf.keras.Sequential([
tf.keras.layers.Conv2D(filters=features,
@@ -166,7 +166,7 @@ class AEEncoderNew(tf.keras.Model):
def __init__(self, layer_params, features, activation='tanh'):
super(AEEncoderNew, self).__init__()
self.encoder = resnet_101(layer_params, features)
self.encoder = resnet_18(layer_params, features)
self.act = tf.keras.layers.Activation(activation)
@@ -193,9 +193,9 @@ class DecoderNew(tf.keras.Model):
self.is_label = is_label
if not is_label:
self.decoder = resnet_101(layer_params, features, reversed=True, is_cvae=is_cvae)
self.decoder = resnet_18(layer_params, features, reversed=True, is_cvae=is_cvae)
else:
self.decoder = resnet_101(layer_params, features, reversed=True, labels=6, is_cvae=is_cvae)
self.decoder = resnet_18(layer_params, features, reversed=True, labels=6, is_cvae=is_cvae)
def call(self, inputs, training):