STDC-Seg icon indicating copy to clipboard operation
STDC-Seg copied to clipboard

Published 20 hours ago • verified publisher icon MichaelFan01

RuntimeError: ONNX export failed: Couldn't export Python operator InPlaceABNSync

Open tianjun-world opened this issue 3 years ago • 5 comments

作者你好，我想转onnx（1.8.0），报这个错，换位requirement要求的配置还是报错，你可以提供onnx代码么

tianjun-world avatar Jul 01 '21 03:07 tianjun-world

latency的测速就是先转onnx哈， 你可以顺着那个代码看一下，你这个错误的话，把InPlaceABNSync换成正常的BatchNorm就行了

MichaelFan01 avatar Jul 01 '21 07:07 MichaelFan01

@MichaelFan01 请问下，InPlaceABNSync转BatchNorm，需要重新训练模型吗？还是训练好的权重也能用？

irvingzhang0512 avatar Aug 13 '21 08:08 irvingzhang0512

@MichaelFan01 请问下，InPlaceABNSync转BatchNorm，需要重新训练模型吗？还是训练好的权重也能用？

理论上，只要relu没设置错的话，应该是可以直接替换的

MichaelFan01 avatar Aug 13 '21 08:08 MichaelFan01

@MichaelFan01 请问下，InPlaceABNSync转BatchNorm，需要重新训练模型吗？还是训练好的权重也能用？

理论上，只要relu没设置错的话，应该是可以直接替换的

感谢，我先查下 abn..

irvingzhang0512 avatar Aug 13 '21 10:08 irvingzhang0512

将model_stages.py中的InPlaceABNSync导入修改为ABN就可以解决：

Francis235 avatar Jul 11 '23 03:07 Francis235