bug fix

2023-01-25 18:04:02 +00:00
parent 2305a0ee23
commit 8514032837
4 changed files with 14 additions and 11 deletions
@@ -175,8 +175,9 @@ class ResNetTypeI(tf.keras.Model):

            x = self.pool1(x)
            x = self.bn1(x, training=training)
-            x = tf.nn.relu(x)
+            # x = tf.nn.relu(x)
            x = self.conv1(x)
+            x = tf.nn.tanh(x)

        return x