pth转onnx遇到以上问题。
解决方案:
将原来的变量换为固定数值解决。
如:
scale=0.5
flow = F.interpolate(flow, scale_factor=scale, mode='bilinear', align_corners=True) * 0.5
改为
flow = F.interpolate(flow, scale_factor=0.5, mode='bilinear', align_corners=True) * 0.5
pth转onnx遇到以上问题。
解决方案:
将原来的变量换为固定数值解决。
如:
scale=0.5
flow = F.interpolate(flow, scale_factor=scale, mode='bilinear', align_corners=True) * 0.5
改为
flow = F.interpolate(flow, scale_factor=0.5, mode='bilinear', align_corners=True) * 0.5