尝试将PyTorch模型导出到ONNX时崩溃: forward()缺少1个必需的位置参数
在将PyTorch模型导出到ONNX时崩溃并出现"forward()缺少1个必需的位置参数"的错误，这通常是由于导出过程中的一些问题导致的。下面是一些可能的原因和解决方法：
- 参数缺失：该错误通常表示在模型的forward()方法中缺少了一个必需的位置参数。在PyTorch中，forward()方法通常接受输入数据作为参数，因此在导出模型时，确保将输入数据正确传递给forward()方法。
- 模型状态：在导出模型之前，确保模型处于正确的状态。这包括确保模型已经被加载并且处于训练或评估模式，以及确保模型的权重和参数已经被正确初始化。
- ONNX版本兼容性：确保使用的PyTorch和ONNX的版本是兼容的。不同版本的PyTorch和ONNX可能存在一些不兼容的问题，因此尽量使用最新的版本，并确保它们之间的兼容性。
- 模型结构和操作的兼容性：某些PyTorch操作可能无法直接转换为ONNX操作，或者ONNX不支持某些特定的模型结构。在导出模型之前，检查模型的结构和使用的操作，确保它们在ONNX中是可行的。
如果以上解决方法无法解决问题，可以尝试以下步骤：
- 更新PyTorch和ONNX的版本，确保使用最新的稳定版本。
- 检查模型的结构和操作，确保它们在ONNX中是支持的。可以查阅PyTorch和ONNX的文档，了解支持的操作和模型结构。
- 尝试使用不同的导出方法或库。除了PyTorch自带的导出功能，还有一些第三方库可以用于将PyTorch模型导出到ONNX，例如torchvision和onnxruntime等。
- 在PyTorch的官方论坛或社区中寻求帮助。如果遇到了无法解决的问题，可以在PyTorch的官方论坛或社区中提问，寻求其他开发者的帮助和建议。
腾讯云相关产品和产品介绍链接地址：
- 腾讯云PyTorch：https://cloud.tencent.com/product/pytorch
- 腾讯云AI引擎：https://cloud.tencent.com/product/tia
- 腾讯云AI Lab：https://cloud.tencent.com/developer/labs/gallery
- 腾讯云AI开放平台：https://ai.qq.com/
相关·内容
我正在使用PyTorch训练神经网络并将它们输出到ONNX中.我在索引中使用这些模型，该索引通过TensorRT加载ONNX。我需要一些特性的一个热编码，但这真的很难在Vespa框架内实现。
是否有可能在我的ONNX网络中(例如，在网络表示之前)嵌入对某些给定功能的单热编码？如果是这样的话，我应该如何基于PyTorch模型来实现这一点？
我已经注意到两件事：
ONNX格式包括OneHot运算符：
PyTorch内置的ONNX导出系统不支持OneHot操作符：
编辑2021/03/11:以下是我的工作流程：
基于PyTorch的学习秩模型的训练
将它们导出为ONNX
浏览 17提问于2021-03-09得票数 3
回答已采纳
我使用Python3.8环境和segmentation_models_pytorch aka smp训练了一个分段模型。当我保存它并在我的预测环境中加载(Python3.6和smp)时，它只在
import torch
model = torch.load(path.join('models', model_name))
但是，它与onnx包(onnx需要更新的Python)发生冲突。我用Python3.10创建了新的conda环境(另一个使用Python3.11)。现在，torch拒绝装载错误消息ModuleNotFoundError: No module named &#
浏览 0提问于2023-05-31得票数 0
1回答
我在PyTorch中有一个PyTorch模型，我想使用CoreML运行它。当将模型导出到ONNX时，输入维度被固定为导出过程中使用的张量的形状，并再次通过从ONNX到CoreML的转换。
import torch
from onnx_coreml import convert
x = torch.ones((32, 1, 1000)) # N x C x W
model = Model()
torch.onnx.export(model, x, 'example.onnx')
mlmodel = convert(model='example.onnx',
浏览 3提问于2020-05-17得票数 4
3回答
我已经通过以下方式将我的模型导出到ONNX：
# Export the model
torch_out = torch.onnx._export(learn.model, # model being run
x, # model input (or a tuple for multiple inputs)
EXPORT_PATH + "mnist.onnx", # where to save the mod
浏览 3提问于2018-11-07得票数 20
有没有人能帮我把YOLOv5 PyTorch模型转换成ONNX或TensorFlow格式，这样就能用OpenCV C++推理了？
我使用本教程通过colab：训练模型
也许在colab中有一种转换方法，或者我应该使用其他工具？
浏览 90提问于2021-09-02得票数 2
3回答
我正在尝试将我的LSTM异常检测Pytorch模型导出到ONNX，但遇到错误。请看我下面的代码。
注意:我的数据形状为2685，5，6。这里是我定义模型的地方：
class Model(torch.nn.Module):
def __init__(self, input_dim, hidden_dim, layer_dim):
super(Model, self).__init__()
self.hidden_dim = hidden_dim
self.layer_dim = layer_dim
self.lstm =
浏览 3提问于2019-08-01得票数 4
背景
torch.mv操作符包含在我的py手电筒模型中，所以当我将该模型转换为onnx时，控制台打印。
RuntimeError: Exporting the operator mv to ONNX opset version 11 is not supported. Please feel free to request support or submit a pull request on Pytorch Github
所以我得为torch.mv操作符做点什么。然而，我无法理解和描述在 of torch.onnx。
问题
我无法确认torch.mv是否是aten运算符，因为我在torc
浏览 8提问于2021-04-02得票数 0
我正在尝试将pyTorch模型转换为onnx，如下所示：
torch.onnx.export(
model=modnet.module,
args=example_input,
f=ONNX_PATH, # where should it be saved
verbose=False,
export_params=True,
do_constant_folding=False,
input_names=['input'],
output_names=['output']
)
modnet是来自以下存储库的模型：
example_
浏览 309提问于2021-02-18得票数 0
回答已采纳
关于轻量服务器windows server 2012升级server2016,问题，在升级过程中重新就蓝屏，有人遇到这个问题的没有啊，这个要怎么解决呢，我用我家里的电脑，就能升级成功，但这个轻量云，就不得行。应该是驱动上边的问题，希望高手解决一下。
浏览 239提问于2023-11-27
1回答
我试图使用onnxjs导入ONNX模型，但我得到了以下错误：
Uncaught (in promise) TypeError: cannot resolve operator 'Cast' with opsets: ai.onnx v11
下面显示了我的html文件中的代码片段。
<html>
<head> </head>
<body>
<!-- Load ONNX.js -->
<script src="https://cdn.jsdelivr.net/npm/onnxjs/di
浏览 7提问于2022-02-25得票数 1
回答已采纳
我想将基于roberta-base的语言模型导出为ONNX格式。该模型使用ROBERTA嵌入并执行文本分类任务。
from torch import nn
import torch.onnx
import onnx
import onnxruntime
import torch
import transformers
来自原木：
17: pytorch: 1.10.2+cu113
18: CUDA: False
21: device: cpu
26: onnxruntime: 1.10.0
27: onnx: 1.11.0
PyTorch出口
batch_size = 3
model_inpu
浏览 66提问于2022-02-22得票数 1
回答已采纳
我正在尝试使用以下命令创建一个带有squeezenet ONNX模型的会话: session = winrt::Windows::AI::MachineLearning::LearningModelSession{模型，winrt::Windows::AI::MachineLearning::LearningModelDevice(deviceKind) }；
我有两个版本的squeezenet。一个在开始时有一个'Sub‘层，而另一个没有。执行上面的代码时，带有“Sub”的代码会抛出错误。
你知道这是怎么回事吗？
浏览 2提问于2018-10-27得票数 1
这将是一个很长的职位，抱歉，提前.
我正在研究去噪算法，我的目标是：
使用PyTorch设计/训练模型，将PyTorch模型转换为CoreML模型
去噪算法由以下三部分组成：
A“下采样”+噪声级map
A正则convnet
An "up-sampling"
第一部分是相当简单的想法，但不容易解释。例如，给出表示图像噪声标准差的输入彩色图像和输入值"sigma“。“向下取样”部分实际上是一个空间到深度的部分。简而言之，对于给定的通道和2x2像素的子集，空间到深度创建了一个由4个通道组成的单个像素。通道数乘以4，而高度和宽度除以2，数据被简单地重新组织。
浏览 1提问于2019-12-04得票数 2
回答已采纳
我遵循，用OpenCV DNN模块加载yolov5*.onnx模型，并使用它进行推理。当试图加载模型时，我得到以下错误：
错误:0@10.376全局E:\Libraries\C++\opencv_gpu\opencv_source\modules\dnn\src\onnx\onnx_importer.cpp (1021) cv::dnn::dnn4_v20220524::ONNXImporter::handleNode DNN/ONNX:在处理节点时出现错误，输入1，输出1:标识：(onnx_node！Identity_0)从域=‘ai.onnx’OpenCV(4.6.0-dev)
E:\L
浏览 44提问于2022-08-01得票数 3
我正在构建一个用于培训模型的API，并认为我希望使用ONNX来来回发送模型。
我正在使用sklearn模型进行测试，在我将其导出到onnx之前，似乎需要对模型进行拟合。
我想要定义一个自定义或标准的滑雪板模型，转换为onnx进行运输、重新打开和训练，保存在ONNX中。
这真的可以吗？我的最终目标是拥有一个API，它可以在未经训练的状态下接受任何sklearn、tensorflow或类似的模型，然后在服务器上进行训练。
浏览 3提问于2022-07-07得票数 0
回答已采纳
2回答
opencv中的负荷onnx模型
、、、、
使用给定的代码在OpenCV++ (4.6.0-dev)中加载onnx模型(从下载)时，我得到以下错误。我做错了什么？请帮我解决这个问题。
错误:0@0.003全局/home/opencv_build/opencv/modules/dnn/src/onnx/onnx_importer.cpp (2564) parseShape DNN/ONNX(形状)：不支持动态“零”形状，输入24300-051全局/home/opencv_build/opencv/modules/dnn/src/onnx/onnx_importer.cpp (1042) handleNode DNN/ONNX:在处理节点
浏览 6提问于2021-10-19得票数 1
我有一个来自联合医院的预科模特。现在，我正尝试使用TensorFlow在python中对该模型进行推断。为此，我使用将ONNX模型保存为SavedModel，以便以后加载该模型。用于保存模型的代码是
import onnx
from onnx_tf.backend import prepare
onnx_model = onnx.load(model_path) # load onnx model
tf_rep = prepare(onnx_model, logging_level='DEBUG')
tf_rep.export_graph(output_path)
加载模型
浏览 4提问于2021-03-24得票数 0
回答已采纳
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
云服务器
ICP备案
实时音视频
即时通信 IM
对象存储活动推荐
腾讯云开发者
