ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Do not store keras model network design if it cannot be serialized

Browse Source
This commit is contained in:
allegroai 2020-01-10 13:41:06 +02:00
parent 073f4c308d
commit 8da8053726
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trains/binding/frameworks/tensorflow_bind.py
View File

@ -1283,7 +1283,11 @@ class PatchKerasModelIO(object):
filepath = kwargs['filepath'] if 'filepath' in kwargs else args[0] filepath = kwargs['filepath'] if 'filepath' in kwargs else args[0]
# this will already generate an output model # this will already generate an output model
config = self._updated_config() try:
config = self._updated_config()
except Exception as ex:
# we failed to convert the network to json, for some reason (most likely internal keras error)
config = {}
# check if object already has InputModel # check if object already has InputModel
if not hasattr(self, 'trains_out_model'): if not hasattr(self, 'trains_out_model'):