Fix: print loss each step

mvpcom · Oct 13, 2017 · 2bb0ad1 · 2bb0ad1
1 parent 2fc1dd1
commit 2bb0ad1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -127,7 +127,7 @@ def train_nn(sess, epochs, batch_size, get_batches_fn, train_op, cross_entropy_l
                                                                      input_image: image,
                                                                      learning_rate: 1e-3,
                                                                      keep_prob: 0.7})
-        print(loss)
+            print(loss)
 tests.test_train_nn(train_nn)