ZFTurbo/Keras-inference-time-optimizer

Investigate slow conversion time

adityapatadia opened this issue · 0 comments

adityapatadia commented

Keras get_weights() and set_weights() function takes most of the time taken in conversion. I could not find a solution but just posting here so someone can find a solution.

Keras-inference-time-optimizer/kito.py

Line 127 in c4e4a78

tmp_model.get_layer(name=layer.name).set_weights(layer.get_weights())