当前位置：网站首页>转换为onnx模型错误汇总

转换为onnx模型错误汇总

2022-08-09 07:50:00 【HySmiley】

1、ValueError: torch.nn.DataParallel is not supported by ONNX exporter, please use 'attribute' module to unwrap model from torch.nn.DataParallel. Try torch.onnx.export(model.module, ...)

多GPU环境下训练的模型在导出onnx文件时，可能会报以上错误。

是因为加载模型前加了此句，model = nn.DataParallel(model)

如果不加，依然报错，如图看到参数字典多了一个module.

 可以遍历模型删除各参数部分中出现的

原网站

版权声明
本文为[HySmiley]所创，转载请带上原文链接，感谢
https://blog.csdn.net/m0_37264397/article/details/123799772

边栏推荐

猜你喜欢

随机推荐