Update custom_objects to reload models
All layers and losses which are not directly implemented by keras have to be passed as custom_objects to keras.load_model.
All layers and losses which are not directly implemented by keras have to be passed as custom_objects to keras.load_model.