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). However, I am having some problems when creating and passing as arguments two dummy inputs for
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!