ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix direct import of keras save/load model functions (issue #355)

Browse Source
This commit is contained in:
allegroai 2021-05-12 15:42:59 +03:00
parent 1e7b35456d
commit 55b214c23d
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1558,6 +1558,12 @@ class PatchKerasModelIO(object):
keras_hdf5.load_weights_from_hdf5_group, PatchKerasModelIO._load_weights)
keras_hdf5.load_weights_from_hdf5_group_by_name = _patched_call(
keras_hdf5.load_weights_from_hdf5_group_by_name, PatchKerasModelIO._load_weights)
if hasattr(keras_hdf5, 'load_model_from_hdf5'):
keras_hdf5.load_model_from_hdf5 = _patched_call(
keras_hdf5.load_model_from_hdf5, PatchKerasModelIO._load_model)
if hasattr(keras_hdf5, 'save_model_to_hdf5'):
keras_hdf5.save_model_to_hdf5 = _patched_call(
keras_hdf5.save_model_to_hdf5, PatchKerasModelIO._save_model)
except Exception as ex:
LoggerRoot.get_base_logger(TensorflowBinding).warning(str(ex))