resnet pyramid

2023-03-02 17:40:14 +00:00
parent 532f5e12f4
commit ec51e5e7cb
7 changed files with 261 additions and 64 deletions
@@ -113,10 +113,10 @@ class Decoder(tf.keras.Model):

 class VariationalEncoderNew(tf.keras.Model):
    
-    def __init__(self, features):
+    def __init__(self, layer_params, features):
        super(VariationalEncoderNew, self).__init__()

-        self.encoder = resnet_18(features)
+        self.encoder = resnet_18(layer_params, features)

        self.mean_branch = tf.keras.Sequential([
            tf.keras.layers.Conv2D(filters=features,