face-parsing.PyTorch
face-parsing.PyTorch copied to clipboard
转onnx出错怎么解决呢?
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.
Also try converting with opset_version=13
https://github.com/ykk648/AI_power/blob/main/face_lib/face_parsing/face_parsing_api.py