Попытка оптимизировать эту простую модель:
from tensorflow import keras
import keras2onnx
from onnx.shape_inference import infer_shapes
from onnx.optimizer import optimize
model = keras.Sequential([
keras.layers.Conv2D(32, (3, 3), activation='relu', input_shape=(28, 28, 1)),
keras.layers.GlobalAveragePooling2D(),
keras.layers.Dense(10)])
model = keras2onnx.convert_keras(model)
model = infer_shapes(model)
model = optimize(model)
приводит к следующей ошибке:
File ".../lib/python3.7/site-packages/onnx/optimizer.py", line 55, in optimize
optimized_model_str = C.optimize(model_str, passes)
IndexError: Input conv2d/kernel:0 is undefined!
Я использую tenorflow = 2.1.0, onnx = 1.7.0 и keras2onnx = 1.6.1.
Есть идеи, как исправить эту проблему?