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

Feature: allow for custom_objects to be fed into load_model #17

Open hexgnu opened 5 years ago

hexgnu commented 5 years ago

Hey there,

I have a model that utilizes a custom loss function and to make this work I changed up the code just a tad to allow for the ability to load custom_objects into the load_model line.

This might be useful to others.