Skip to content

Missing weights input when weights is not a constant input for paddedConv

When testing a model with paddedConv metaop, I get the error: RuntimeError: Missing input#1 for operator Conv2D when compiling the model.
This only happends when input#1 (weights) is not a constant.
Here is the model conv_with_padding.onnx And the test script:

import aidge_core;
import aidge_backend_cpu;
import aidge_onnx
import numpy as np

model = aidge_onnx.load_onnx("conv_with_padding.onnx")

X = np.random.randn(1, 1, 5, 5).astype(np.float32)
W = np.random.randn(1, 1, 3, 3).astype(np.float32)
x_tensor= aidge_core.Tensor(X)
w_tensor= aidge_core.Tensor(W)

model.compile("cpu", aidge_core.dtype.float32, dims=[[1, 1, 5, 5],[1, 1, 3, 3]])
scheduler = aidge_core.SequentialScheduler(model)

# Run inference !
scheduler.forward(data=[x_tensor, w_tensor])
for outNode in model.get_output_nodes():
	output_aidge = np.array(outNode.get_operator().get_output(0))
	print(output_aidge)
Edited by Pierre Gaillard