clear session only for tf

levinas · levinas · commit 141c13193e3a · 2017-01-25T17:26:14.000-07:00
diff --git a/P1B3/p1b3_baseline.py b/P1B3/p1b3_baseline.py
@@ -378,4 +378,7 @@ def main():
 
 if __name__ == '__main__':
     main()
-    K.clear_session()
+    try:
+        K.clear_session()
+    except AttributeError:      # theano does not have this function
+        pass
