onnx KeyError: 'upsample_bilinear2d'

 

onnx KeyError: 'upsample_bilinear2d'

 

解決方法：把上採樣的bilinear換成可以支持的。

比如：

import torch.nn.functional as funtion

funtion.interpolate(P3_out, size=P4_td.shape[2:], mode="nearest"))