Hello,
I am trying to export a torchvision.model.
the model is r3d-18 (resnet3D)
import torch
import torch.onnx
import torchvision
dummy_input = torch.randn(1, 3, 4, 112, 112)
model = torchvision.models.video.r3d_18(pretrained=True)
torch.onnx.export(model, x, "ResNet3D-18.onnx")
I tried to load the onnx file on matlab.
but matlab say ‘mismatch size error, It is different with expected size’.
I found the input shape at https://github.com/pytorch/pytorch/pull/30121/files