towards fixing #318

uclnlp · Oct 29, 2017 · 543ebff · 543ebff
1 parent 4dc7b5e
commit 543ebff
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/jack/debug/base.py b/jack/debug/base.py
@@ -21,3 +21,9 @@ def test_update(feed_dict, train_op):
         res |= (b != a).any()
 
     return res
+
+
+def test_loss(feed_dict, loss_op):
+    session = tf.Session()
+    loss = session.run(loss_op, feed_dict=feed_dict)
+    return loss != 0