search

DwangoMediaVillage / keras_compressor

Model Compression CLI Tool for Keras.
https://nico-opendata.jp/ja/casestudy/model_compression/index.html
157 stars 39 forks source link

Make layers.py and utils.py compatible with keras 2.2 #11

Closed keineahnung2345 closed 1 year ago

keineahnung2345 commented 6 years ago

layers.py: Keras 2.2 use layer._inbound_nodes, layer._outbound_nodes instead of layer.inbound_nodes, layer.outbound_nodes. utils.py conv_utils.normalize_data_format should be replaced with K.normalize_data_format in keras 2.2.

Ref: ekholabs/keras-contrib@0dac2da#diff-0970ecd5f755557421e7ba7716bdfd6a https://github.com/apple/coremltools/blob/master/coremltools/converters/keras/_topology2.py