Torch.onnx.export() does not support torch.val_mean() op


I’m trying to convert my model to onnx format but the val_mean op is not supported.
My pytorch version is 1.2.

I know that upgrading to pytorch 1.8 could solve the issue but I have to use 1.2 for some reason.

How to solve this on pytorch 1.2? Any suggestions will be appreciated.


still looking for help…