run train_step and eval step separately (#215)

lsds · Nov 16, 2019 · c234715 · c234715
1 parent 6f22b4f
commit c234715
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/srcs/python/kungfu/tensorflow/v1/examples/__main__.py b/srcs/python/kungfu/tensorflow/v1/examples/__main__.py
@@ -34,12 +34,16 @@ def sgd_example():
     with tf.Session() as sess:
         sess.run(init)
         for step in range(5):
-            v, _ = sess.run([x, train_step])
+            # v, _ = sess.run([x, train_step]) # result is not deterministic!
+            sess.run(train_step)
+            v = sess.run(x)
             print('step %d, result: %f' % (step, v))
 
             u = (1 - 2 * lr)**(step + 1)
             if abs(u - v) > 1e-6:
-                print('unexpected result: %f, want: %f' % (v, u))
+                msg = 'unexpected result: %f, want: %f' % (v, u)
+                print(msg)
+                raise RuntimeError(msg)
 
 
 def main(args):