fix load_model so that it can load customized optimizer (keras-team#4625

)
ahlfors · Dec 7, 2016 · 1fd2108 · 1fd2108
1 parent ad5e29a
commit 1fd2108
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/models.py b/keras/models.py
@@ -151,7 +151,7 @@ def deserialize(obj):
         return model
     training_config = json.loads(training_config.decode('utf-8'))
     optimizer_config = training_config['optimizer_config']
-    optimizer = optimizer_from_config(optimizer_config)
+    optimizer = optimizer_from_config(optimizer_config, custom_objects=custom_objects)
 
     # recover loss functions and metrics
     loss = deserialize(training_config['loss'])