face-parsing.PyTorch icon indicating copy to clipboard operation
face-parsing.PyTorch copied to clipboard

转onnx出错怎么解决呢?

Open 205418367 opened this issue 3 years ago • 3 comments

205418367 avatar Dec 20 '21 10:12 205418367

I convert successfully by replacing F.avg_pool2d with F.adaptive_avg_pool2d in models.py. And pytorch version is 1.12.1 and onnx-pytorch version is 0.1.5 in my environment.

JinShiyin avatar Sep 05 '22 12:09 JinShiyin

Also try converting with opset_version=13

https://github.com/ykk648/AI_power/blob/main/face_lib/face_parsing/face_parsing_api.py

ykk648 avatar Jul 07 '23 09:07 ykk648