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

升级pytroch >1.3

opset_version=10

torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11)

你可能感兴趣的:(杂货)