disable model plot on server

TianyuDu · Oct 9, 2018 · ed528d6 · ed528d6
1 parent 30548f5
commit ed528d6
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 5 deletions.
diff --git a/k models/exchange/__pycache__/multi_config.cpython-36.pyc b/k models/exchange/__pycache__/multi_config.cpython-36.pyc
diff --git a/k models/exchange/core/containers/__pycache__/base_container.cpython-36.pyc b/k models/exchange/core/containers/__pycache__/base_container.cpython-36.pyc
diff --git a/k models/exchange/core/containers/__pycache__/multivariate_container.cpython-36.pyc b/k models/exchange/core/containers/__pycache__/multivariate_container.cpython-36.pyc
diff --git a/k models/exchange/core/models/__pycache__/base_model.cpython-36.pyc b/k models/exchange/core/models/__pycache__/base_model.cpython-36.pyc
diff --git a/k models/exchange/core/models/__pycache__/multivariate_lstm.cpython-36.pyc b/k models/exchange/core/models/__pycache__/multivariate_lstm.cpython-36.pyc
diff --git a/k models/exchange/core/models/multivariate_lstm.py b/k models/exchange/core/models/multivariate_lstm.py
@@ -193,11 +193,14 @@ def save_model(
 
         # Save model illustration to png file.
         print("Saving model visualization...")
-        keras.utils.plot_model(
-            self.core,
-            to_file=f"{folder}model.png",
-            show_shapes=True,
-            show_layer_names=True)
+        try:
+            keras.utils.plot_model(
+                self.core,
+                to_file=f"{folder}model.png",
+                show_shapes=True,
+                show_layer_names=True)
+        except:
+            print("Model illustration cannot be saved.")
 
         # Save training history (if any)
         if self.hist is not None:

diff --git a/k models/exchange/core/tools/__pycache__/bokeh_visualize.cpython-36.pyc b/k models/exchange/core/tools/__pycache__/bokeh_visualize.cpython-36.pyc