diff options
Diffstat (limited to 'tensorflow/python/keras/engine/saving.py')
-rw-r--r-- | tensorflow/python/keras/engine/saving.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/keras/engine/saving.py b/tensorflow/python/keras/engine/saving.py index a2eed7cb46..a2f31fda8f 100644 --- a/tensorflow/python/keras/engine/saving.py +++ b/tensorflow/python/keras/engine/saving.py @@ -248,7 +248,7 @@ def load_model(filepath, custom_objects=None, compile=True): # pylint: disable= loss = convert_custom_objects(training_config['loss']) metrics = convert_custom_objects(training_config['metrics']) weighted_metrics = convert_custom_objects( - training_config['weighted_metrics']) + training_config.get('weighted_metrics', None)) sample_weight_mode = training_config['sample_weight_mode'] loss_weights = training_config['loss_weights'] |