我正在使用可变批次大小(输入: None,320,240,3)在Tensorflow中训练一个模型。问题是在训练后的量化过程中,我不能有任何动态输入,因此没有“无”,而使用edgetpu编译器时,批大小不能大于1。
我目前的方法是再训练一次,固定的批次为1,但这有点乏味。
是否可以将投入从零、320、240、3改为1,320、240、3或320、240、3,而不必再进行一次培训?
发布于 2022-09-12 08:08:31
对我有用的一个解决方法是创建一个具有特定批次大小的类似模型,并从原始模型复制权重。
https://stackoverflow.com/questions/69541719
复制相似问题