search

ZFTurbo / Keras-inference-time-optimizer

Optimize layers structure of Keras model to reduce computation time
MIT License
157 stars 18 forks source link

Reduced Model Export #20

Closed JPery closed 5 years ago

JPery commented 5 years ago

Is there any way to export the model so we can avoid converting the model each time?

ZFTurbo commented 5 years ago

Sure. You can just save it as any other Keras model:

from kito import reduce_keras_model
model_reduced = reduce_keras_model(model)
model_reduced.save(output_path)
JPery commented 5 years ago

Thank you for your fast answer.

This is a nice project. It reduced mi inference time from 120 to 50 seconds. Great job!