根据错误信息,导出模型时遇到了不支持的输入类型:numpy.ndarray
。
torch.onnx.export()
函数要求输入和输出是元组、列表或变量(Variables)的形式。它不支持字典或字符串作为输入和输出。
您可以尝试将输入数据转换为列表或元组,并再次运行导出代码。例如:
input_data = [your_input_data] # 将numpy数组转换为列表
torch.onnx.export(model, input_data, "model.onnx")
如果您仍然遇到问题,请检查代码中是否有其他地方使用了不受支持的数据类型,并尝试进行相应修改以符合torch.onnx.export()
函数的要求。
内容由零声教学AI助手提供,问题来源于学员提问