diff options
author | ZJaume <jzaragoza@prompsit.com> | 2022-03-09 19:09:17 +0300 |
---|---|---|
committer | ZJaume <jzaragoza@prompsit.com> | 2022-03-09 19:09:17 +0300 |
commit | fe01ea476dd22cc20feab8a517c6ea52ef091241 (patch) | |
tree | 72392cffbbf813368466e3254b807dbfd14f68dc | |
parent | 7eaf609c6504f884eb32061d09ec5658b19f0453 (diff) |
Add Keras backend alias to custom_objs when loading lite models
-rw-r--r-- | bicleaner_ai/models.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/bicleaner_ai/models.py b/bicleaner_ai/models.py index 0f2c914..e06c162 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -259,6 +259,7 @@ class BaseModel(ModelInterface): deps = {'FScore': FScore, 'MatthewsCorrCoef': MatthewsCorrCoef, 'TokenAndPositionEmbedding': TokenAndPositionEmbedding, + 'K': tf.keras.backend, } # Try loading the whole model |