ONNX 변환 시 "ONNX export failed: Couldn't export operator aten::upsample_bilinear2d" 에러

Peloton·2023년 10월 31일
0
post-custom-banner

Issue

ONNX 변환 시 아래 에러가 발생하는 경우!

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

Solution

opset 버전을 11로 변경한다. (저의 경우 9로 되어 있었음.)

torch.onnx.export(..,opset_version=11)
profile
Software Engineeeeer

0개의 댓글