towards fixing #318

uclnlp · Oct 29, 2017 · 4dc7b5e · 4dc7b5e
1 parent 89c26d2
commit 4dc7b5e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/jack/debug/__init__.py b/jack/debug/__init__.py
@@ -0,0 +1 @@
+# -*- coding: utf-8 -*-
diff --git a/jack/debug/base.py b/jack/debug/base.py
@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+
+import tensorflow as tf
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+
+def test_update(feed_dict, train_op):
+    session = tf.Session()
+    session.run(tf.global_variables_initializer())
+
+    before = session.run(tf.trainable_variables())
+    session.run(train_op, feed_dict=feed_dict)
+    after = session.run(tf.trainable_variables())
+
+    res = False
+    for b, a in zip(before, after):
+        # Check if anything changed
+        res |= (b != a).any()
+
+    return res