Tag Archives: ONNX export failed

RuntimeError: ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d

pth –> Onnx error

RuntimeError: ONNX export failed: Couldn't export operator aten::upsample_bilinear2d

Solution:

   torch.onnx.export(model, input, onnx_path, verbose=True, input_names=input_names, output_names=output_names)
#add:opset_version=11
   torch.onnx.export(model, input, onnx_path, verbose=True, input_names=input_names, output_names=output_names, opset_version=11)