diff options
| author |  Sergii Khomenko <sergii.khomenko@stylight.com> | 2018-09-05 01:06:21 +0200 |
|---|---|---|
| committer | Sergii Khomenko <sergii.khomenko@stylight.com> | 2018-09-05 01:34:33 +0200 |
| commit | 462f1871ee405ba7184a6d4c113d15b764e80324 (patch) | |
| tree | 77228c4de5d6148dd4adcb8362fecc0f39a2e9bd | |
| parent | 0065d3389a63a529469dc71e950c66da2ebdbc24 (diff) | |
Add an explicit reason for NotImplementedError on eager model save
| -rw-r--r-- | tensorflow/python/keras/engine/network.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/keras/engine/network.py b/tensorflow/python/keras/engine/network.py index cd74e36e68..f8c23ed124 100644 --- a/tensorflow/python/keras/engine/network.py +++ b/tensorflow/python/keras/engine/network.py @@ -1355,7 +1355,9 @@ class Network(base_layer.Layer): ``` """ if not self._is_graph_network: - raise NotImplementedError + raise NotImplementedError( + 'Currently `save` requires model to be a graph network. Consider ' + 'using `save_weights`, in order to save the weights of the model.') from tensorflow.python.keras.models import save_model # pylint: disable=g-import-not-at-top save_model(self, filepath, overwrite, include_optimizer) |