因为我的torch 是1.2的,所以不支持 orch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11),pytorch >= 1.3.0才支持opset_version=11
最快的方式是直接将mode配置的bilinear换成nearest
因为我的torch 是1.2的,所以不支持 orch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11),pytorch >= 1.3.0才支持opset_version=11
最快的方式是直接将mode配置的bilinear换成nearest
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。