search

ZFTurbo / Keras-inference-time-optimizer

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

Model is not correct when has multiple inputs and outputs #3

Closed mrlzla closed 6 years ago

mrlzla commented 6 years ago

At the last line of reduce_keras_model method one creates model that have only one input and one output, it's not true in general since keras model can get multiple inputs and multiple outputs.

ZFTurbo commented 6 years ago

Thank you. I guess we need to create some generic test for such case.

I also know the problem with complex models which included other models as layers. Code won't work on them as well. For example: RetinaNet.

ZFTurbo commented 6 years ago

The problem was fixed. Test with multi input and multi output was added.

mrlzla commented 6 years ago

Good job!