Merge pull request #233 from jonasrauber/graph_assertion

catch graph mismatch earlier
bethgelab · Nov 5, 2018 · 6a7628e · 6a7628e
2 parents 9ce045b + 4b0932e
commit 6a7628e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/foolbox/models/tensorflow.py b/foolbox/models/tensorflow.py
@@ -50,6 +50,8 @@ def __init__(
             self._created_session = True
         else:
             self._created_session = False
+            assert session.graph == images.graph, \
+                'The default session uses the wrong graph'
 
         with session.graph.as_default():
             self._session = session