Switched from SGD to Adam Optimizer

Alpslee · Oct 3, 2016 · 61ea801 · 61ea801
1 parent cc518e6
commit 61ea801
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -8,7 +8,7 @@
 sess = tf.InteractiveSession()
 
 loss = tf.reduce_mean(tf.square(tf.sub(model.y_, model.y)))
-train_step = tf.train.GradientDescentOptimizer(0.01).minimize(loss)
+train_step = tf.train.AdamOptimizer(1e-4).minimize(loss)
 sess.run(tf.initialize_all_variables())
 
 saver = tf.train.Saver()