Initial

fjnunes · Apr 25, 2016 · 17f0874 · 17f0874
1 parent 58c1781
commit 17f0874
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/fully_connected.py b/fully_connected.py
@@ -68,7 +68,7 @@ def loss(logits, labels):
   cross_entropy = tf.nn.softmax_cross_entropy_with_logits(logits,
                                                           onehot_labels,
                                                           name='xentropy')
-  loss = tf.add(tf.reduce_mean(cross_entropy, name='xentropy_mean'), tf.nn.l2_normalize(logits, dim=1))
+  loss = tf.reduce_mean(cross_entropy, name='xentropy_mean')
   return loss
 
 

diff --git a/output_graph.pb b/output_graph.pb
diff --git a/train.py b/train.py
@@ -20,7 +20,7 @@
 flags = tf.app.flags
 FLAGS = flags.FLAGS
 flags.DEFINE_float('learning_rate', 0.01, 'Initial learning rate.')
-flags.DEFINE_integer('max_steps', 40000, 'Number of steps to run trainer.')
+flags.DEFINE_integer('max_steps', 20000, 'Number of steps to run trainer.')
 flags.DEFINE_integer('hidden1', 128, 'Number of units in hidden layer 1.')
 flags.DEFINE_integer('hidden2', 32, 'Number of units in hidden layer 2.')
 flags.DEFINE_integer('batch_size', 100, 'Batch size.  '