Hello,
I am currently working on a project that relies on a Convolutional Neural Network that has two different inputs(two images).
My idea is to obtain some information about the network and I need to export it through torch.onnx.export (ONNX Export 13). However, I am having some problems when creating and passing as arguments two dummy inputs for torch.onnx.export.
Does any one know if this function supports CNNs with two inputs?
My code is something as follows:
dummy_input1 = torch.randn(1, 3, 224, 224).cuda()
dummy_input2 = torch.randn(1, 150, 224, 224).cuda()
torch.onnx.export(model, (dummy_input1, dummy_input2))

Thanks in advance!
Alex.