Resnet

jw787 · Oct 12, 2019 · 094c7a1 · 094c7a1
1 parent aeea074
commit 094c7a1
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/3-Neural_Network_Architecture/Resnet/Resnet.py b/3-Neural_Network_Architecture/Resnet/Resnet.py
@@ -88,7 +88,6 @@ def call(self, x, training=False):
         out = self.avg_pool(out)
         out = tf.reshape(out, (out.shape[0], -1))
         out = self.fc(out)
-        # return tf.argmax(tf.nn.softmax(out, axis=-1), axis=-1)
         return tf.nn.softmax(out, axis=-1)
 
 # Load and prepare the cifar10 dataset.