From 543ebff1ed971a1b9b3f8fed02a945939e50dc45 Mon Sep 17 00:00:00 2001
From: Pasquale Minervini <pasquale.minervini@insight-centre.org>
Date: Sun, 29 Oct 2017 18:10:06 +0100
Subject: [PATCH] towards fixing #318

---
 jack/debug/base.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/jack/debug/base.py b/jack/debug/base.py
index 6220c003..4b1b8a42 100644
--- 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