keras的model通常使用的时候是不需要特意把input_shape设置明白的,但今天用tf-lite部署到MCU上的时候,报错

None is only supported in the 1st dimension.……………………<\p> 就是说只有1st dimentio(也就是batch_size)可以为None

那就改呗
结果model.input_size = (None, 128, 2)会报错说不给改。。
然后google了一个解决方案,如下

from tensorflow.keras.layers import Input
from tensorflow.keras.models import Model

newInput = Input(shape=(128,2))
newOutputs = model(newInput)
newModel = Model(newInput,newOutputs)
newModel.set_weights(model.get_weights()) #可一定注意把weights换咯