update curvature CVAE

2022-09-22 11:26:07 +00:00
parent aa1e7be0e6
commit c5ae37aacc
3 changed files with 46 additions and 34 deletions
@@ -369,7 +369,7 @@ class ConvNeXtReverseLabels(tf.keras.Model):
        self.last = Dense(nr_of_classes, activation='softmax')

    def call(self, x):
-        for i in reversed(len(depths)):
+        for i in reversed(range(len(self.downsample_layers))):
            x = self.downsample_layers[i](x)
            x = self.stages[i](x)
        x = self.last(x)