Traceback (most recent call last):


 "test_coreml.py", line 83, in <module>

custom_conversion_functions={ "MaxPoolingWithArgmax2D": ConvertMaxPoolingWithArgmax2D } )


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/coremltools/converters/keras/_keras_converter.py", line 804, in convert

use_float_arraytype=use_float_arraytype)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/coremltools/converters/keras/_keras_converter.py", line 585, in convertToSpec

use_float_arraytype=use_float_arraytype)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/coremltools/converters/keras/_keras2_converter.py", line 319, in _convert

model = _keras.models.load_model(model, custom_objects = custom_objects)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/engine/saving.py", line 419, in load_model

model = _deserialize_model(f, custom_objects, compile)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/engine/saving.py", line 225, in _deserialize_model

model = model_from_config(model_config, custom_objects=custom_objects)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/engine/saving.py", line 458, in model_from_config

return deserialize(config, custom_objects=custom_objects)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/layers/__init__.py", line 55, in deserialize

printable_module_name='layer')


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/utils/generic_utils.py", line 145, in deserialize_keras_object

list(custom_objects.items())))


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/engine/network.py", line 1022, in from_config

process_layer(layer_data)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/engine/network.py", line 1008, in process_layer

custom_objects=custom_objects)


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/layers/__init__.py", line 55, in deserialize

printable_module_name='layer')


 "/Users/carlos/anaconda3/envs/ml/lib/python3.7/site-packages/keras/utils/generic_utils.py", line 138, in deserialize_keras_object

': ' + class_name)
ValueError: Unknown layer: MaxPoolingWithArgmax2D

`Traceback (most recent call last):


 "inference.py", line 26, in <module>

prediction_model.load_weights(model_path, by_name=True)


 "/usr/local/lib/python3.7/dist-packages/tensorflow_core/python/keras/engine/training.py", line 182, in load_weights

return super(Model, self).load_weights(filepath, by_name)


 "/usr/local/lib/python3.7/dist-packages/tensorflow_core/python/keras/engine/network.py", line 1371, in load_weights

saving.load_weights_from_hdf5_group_by_name(f, self.layers)


 "/usr/local/lib/python3.7/dist-packages/tensorflow_core/python/keras/saving/hdf5_format.py", line 754, in load_weights_from_hdf5_group_by_name

str(weight_values[i].shape) + '.')
ValueError: Layer #1097 (named "class_head"), weight <tf.Variable 'pyramid_classification/kernel:0' shape=(3, 3, 384, 180) dtype=float32> has shape (3, 3, 384, 180), but the saved weight has shape (720, 384, 3, 3).
