Problem with transferring a model from PyTorch to Caffe2

Hi everyone,

I am trying to follow the instructions in the pytorch tutorial for transferring a pytorch model to Caffe2. But I get the following error when trying to execute: prepared_backend = onnx_caffe2.backend.prepare(model)

“onnx.onnx_cpp2py_export.checker.ValidationError: Input size 2 not in range [min=1, max=1]”