search

MPolaris / onnx2tflite

Tool for onnx->keras or onnx->tflite. Hope this tool can help you.
Apache License 2.0
526 stars 42 forks source link

ValueError: Layer p_re_lu weight shape (1, 1, 32) is not compatible with provided weight shape (32, 1, 1). #47

Closed liziru closed 1 year ago

liziru commented 1 year ago

image image My model is a unet+crn arch, any advice to solve this error?

MPolaris commented 1 year ago

you can debug layers\activations_layers.py #63-#75 lines.

liziru commented 1 year ago

you can debug layers\activations_layers.py #63-#75 lines.

感谢，已解决，但发现KeyError: 'LSTM not implemented yet'问题，有没有支持lstm、gru的计划啊？

MPolaris commented 1 year ago

目前没有计划，你可以自己尝试添加。