为了标准化的目的，我想在vgg16网络的基础上创建一个网络，但是在conv2d层之后添加线性层(Gemm)。之后，我想将网络导出为ONNX文件。第一部分似乎是可行的:我使用Pytorch代码来生成vgg16，并按如下方式修改它 import torch.nn as nn
"&quo
浏览 73提问于2020-12-08得票数 1
1回答
我有一个python程序，在.py文件中有以下代码片段，它使用trtexec将ONNX模型转换为TRT引擎： subprocess.run([sys.executable, "-c", "trtexec --onnx=rmpx.onnx --saveEngine=rmpx_engine_pytorch.trt --explicitBatch --fp16"]) subprocess.run([s
浏览 6提问于2021-06-13得票数 0
我尝试使用ONNX将MATLAB模型转换为PyTorch，就像Andrew在这里提出的那样：
import onnx
pytorch_model
浏览 1提问于2021-02-21得票数 0
回答已采纳
1回答
预期行为在python中，ONNX和PyTorch模型产生相同的输出。但是，在Barracuda中运行的相同的ONNX</e
浏览 3提问于2020-07-02得票数 1
回答已采纳
当使用GPT2时，我们可以简单地传递'labels‘参数来获得损失，如下所示： import torchoutputs = model(**inputs, labels=inputs["input_ids"])
loss = outputs.loss 但是，不能找出如何在ONNX推理会话中获得相同的损失。我使用</
浏览 88提问于2021-08-10得票数 0
目标:重新开发这个以使用。我将PyTorch / HuggingFace变压器模型转换为ONNX并存储它。DecodeError发生在onnx.load()上。。
问题不是..。Quantis
浏览 75提问于2022-01-20得票数 0
回答已采纳
1回答
如何手动创建ONNX文件？我的意思是，如果我们不使用像PyTorch，caffe2，e.t.c这样的框架，我们可以(二进制编码)创建一个onnx文件吗，如果我们知道之前的网络细节，比如节点的#，节点的类型，它们的连接，e.t.c
浏览 10提问于2019-07-18得票数 2
回答已采纳
我有一个用pytorch训练的神经网络，我想把它部署到Unity应用程序中。做这件事最好的方法是什么？我还对允许用户在Unity应用程序中进一步训练神经网络感兴趣，我猜这将需要将pytorch的某些部分集成到Unity中(也许有一种方法可以将pytorch的C++ / torchscript API与Unity如果有人有这方面的经验，我想知道最好的替代方案是什么。
浏览 5提问于2019-05-27得票数 4
1回答
我使用PyTorch训练了一个YOLOv3-SPP模型。然后，我将模型保存为onnx格式，然后使用onnx-coreml将我的onnx模型转换为CoreML。我似乎不知道如何使用我的模型，因为输出是MLMultiArrays。
我是机器学习的新手，当尝试使用这个模型时，我不知道从哪里开始。我不知道每个MultiArray包含</
浏览 4提问于2020-03-14得票数 0
1回答
我正在尝试将PyTorch模型(包含权重的pth文件)转换为onnx文件，然后再转换为TensorFlow模型，因为我使用的是TensorFlow。对其进行微调。到目前为止，这是我的尝试。enter image description here，我认为问题是权重是用于视觉转换器的。但是我还没有弄清楚用什么类型的模型来转换它。我假设有一个CRNN，但如果有更简单的方法，我很想知道。P
浏览 102提问于2021-08-18得票数 0
回答已采纳
我做了一个简单的pytorch MLP(GAN生成器)，并使用教程()将其转换为onnx，我的代码有点不同，但我无法捕捉到错误。torch.save(G.state_dict(), "pytorch_model.pth") at new Promise (<anonym
浏览 23提问于2020-05-22得票数 0
回答已采纳
1回答
我正在试着为《朱莉娅》中的onnx模特做一个读者。通常，onnx模型由一个model.pb文件组成，该文件包含有关模型的元数据。但是，我不知道如何使用协议缓冲区或协议缓冲区的Julia实现(ProtoBuf.jl)从这个model.pb文件生成数据结构，以便稍后可以为模型生成Julia代码。谁能告诉我怎样才能让一个阅读器将onnx模型加载到Jul
浏览 3提问于2018-01-27得票数 1
1回答
我正在尝试检查我的.onnx模型是否正确，并且需要运行推断来验证相同的输出。我找到了MXNet框架，但我似乎无法理解如何导入模型--我只有.onnx文件，除了onnx模型之外，MXNet还需要额外的输入。在Python中还有其他简单<em
浏览 3提问于2018-06-21得票数 1
回答已采纳
我将我的tensorflow模型转换为具有tensorRT精度模式的FP16。而且速度比我预期的要慢。在优化之前，我使用.pb冻结图进行推理。经过tensorRT运算后，我得到了14 have。以下是Jetson网站的基准测试结果
下面是使用TensorRt优化tensorflow保存模型的
浏览 10提问于2021-02-07得票数 6
回答已采纳
我读过许多关于如何使用PyTorch对数据集进行回归的教程，例如，使用由几个线性图层和均方误差损失组成的模型。 好吧，假设我知道函数F依赖于变量x和一些未知参数(p_j: j=0，...，P-1)，P相对较小，但该函数是一个特殊函数的组合。因此，我的问题是已知数据{x_i，y_i}_i<=N的经典最小化 Min_{ {p_j} } Sum_i (F(x_i;{p_j}) - y_i)^2 我想
浏览 14提问于2020-01-13得票数 0
1回答
我很好奇，也不知道你是怎么救你训练过的模特的？有点像python中的泡菜文件，但是我正在考虑保存到ONNX中，但是我不知道如何通过我自己的训练和测试功能手动完成它。我试着搜索如何在google中保存机器学习模型，但是所有的结果都显示了如何在使用泡菜的python上进行保存。我有预感将数据保存为二进制文件？但我无法想象我试图复制的训练模型中的数据。
浏览 10提问于2022-11-16得票数 2
回答已采纳
3回答
我有一个用Python和Keras实现的神经网络。一旦我训练了它，我就导出了模型，我得到了两个文件: model.js和model.h5。现在，我想在.NET项目中进行实时分类，并使用经过训练的神经网络进行分类。 在ML.NET中，有没有一种方法可以把用python导出的模型和经过训练的权重加载到模型对象中？我在documentation1中看到可以加载以前保存的模型，但显然是存储在.zip中，我找不到格式(可能是为了编写一个脚本，从python中获取模型并将其‘转换’
浏览 22提问于2019-04-30得票数 4
回答已采纳
我使用 PyTorch重新实现训练了一个PyTorch代理，并将该代理保存为.pth。现在，我需要将代理转换成.onnx格式，我正在使用这个this：import torchvisionhas no attribute 'state_dict' (下面是完整的提示)。据我所知，问题是将代理转换为.onnx需要更多信息。我是
浏览 0提问于2020-03-25得票数 0
回答已采纳
我正在使用Tensorflow.js创建一个程序。它应该接收一个onnx文件，并能够使用tf加载该文件，从而能够进行推断。 我的问题是如何将其从onnx转换为tfjs？我宁愿只使用js来解决这个问题。(所以我不能使用https://github.com/onnx/onnx-tensorflow，因为python) 如果有帮助，文件</
浏览 82提问于2021-07-05得票数 0
云服务器
ICP备案
云直播
对象存储
实时音视频
腾讯云开发者
