majianjia / nnom

A higher-level Neural Network library for microcontrollers.
Apache License 2.0
908 stars 245 forks source link

Conv1D layer after RNN layer incur an error at inference. #180

Closed bfs18 closed 1 year ago

bfs18 commented 1 year ago

The output of a RNN layer is a 3-dim tensor, but the input of the Conv1D layer is a 4-dim tensor. This cause an error at inference if I put a Conv1D layer after RNN layer. Honestly, I use the Conv1D layer with kernel size 1 instead of a Dense layer as the output layer, so per channel quantization can be applied. Are there any solutions to fix this?