在R包" Torch“中定义torch类
在R包"Torch"中，定义了torch类。torch类是一个用于深度学习的框架，它提供了一系列的函数和工具，用于构建和训练神经网络模型。
torch类的主要特点和优势包括：
- 强大的计算能力：torch类基于高效的张量计算库，能够快速进行矩阵运算和数值计算，支持并行计算和GPU加速，能够处理大规模的数据集和复杂的模型。
- 灵活的模型构建：torch类提供了丰富的模型构建工具，可以通过简单的函数调用来定义神经网络的结构，支持各种常见的网络层和激活函数，同时也支持自定义的网络结构。
- 自动求导：torch类内置了自动求导功能，可以自动计算模型参数的梯度，简化了模型训练过程中的梯度计算和反向传播操作。
- 丰富的预训练模型：torch类提供了一系列的预训练模型，包括常用的图像分类、目标检测和语义分割模型等，可以方便地进行迁移学习和模型微调。
- 多样化的应用场景：torch类适用于各种深度学习任务，包括图像处理、自然语言处理、推荐系统等，可以应用于科研、工业生产和商业应用等领域。
推荐的腾讯云相关产品和产品介绍链接地址：
- 腾讯云AI Lab：https://cloud.tencent.com/product/ai-lab
- 腾讯云机器学习平台：https://cloud.tencent.com/product/tensorflow
- 腾讯云深度学习工具包：https://cloud.tencent.com/product/dltp
相关·内容
1回答
这篇文章与我之前的How to define a Python Class which uses R code, but called from rTorch?有关。 我偶然发现了R (https://torch.mlverse.org/docs/index.html)中的torch包，它允许定义DataSet类定义。但是，我还需要能够在Python语言中定义像class MyModelClass(torch.nn.Module)这样的模型类。这在R中的torch包中是可能的吗？ 当我尝试用reticulate来做这件事时，它不起作用--有一些冲突，比如 ImportError: /Us
浏览 95提问于2020-12-24得票数 3
回答已采纳
我正在尝试安装torch v1.5.1，CPU版本，以及torchvision。 运行pip install torch==1.5.1+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html运行良好， 但是我不能从包含这两行代码的requirements.txt安装： torch==1.5.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.6.1+cpu -f https://downlo
浏览 23提问于2020-09-02得票数 0
回答已采纳
我一直试图通过vm实例上的控制台将我的模型部署到AI平台上进行预测，但是我得到了错误"(gcloud.beta.ai-platform.versions.create)创建版本失败。错误检测到的坏模型：“加载模型失败:加载模型时出现意外错误:ModuleNotFoundError中的问题:没有名为'torchvision‘的模块(错误代码: 0)”
我需要同时包括torch和torchvision。我遵循了这个问题中的步骤，但无法获取用户gogasca所指向的文件。我试着从Pytorch网站下载 .whl文件并将其上传到我的云存储中，但得到的错误与没有模块torchvision
浏览 1提问于2020-05-21得票数 2
回答已采纳
1回答
火炬的OOP结构
、、
我已经开始学习OOP来实现我的DL模型。我在哪里可以找到毕火炬的OOP结构。所谓结构，我指的是类结构(即类的继承结构，如顺序容器)。我知道模块类是大多数结构的基础。
此外，我在github上探索Py手电，我也很困惑Py火炬中的导入语句是如何工作的。例如，模块类是在torch/nn/modules/module.py中定义的，但我们只导入torch.nn并编写nn.Module来表示模块类。按其存储在github上的方式，它不应该是nn.modules.module.Module吗？
# Example of using Sequential
model = nn.Sequential(
nn.
浏览 2提问于2021-05-24得票数 1
回答已采纳
1回答
在python中，如果只导入torch (而不是导入numpy)，"torch.numpy()“仍然可以工作。这是否意味着numpy数据可以在没有numpy包的情况下存储和显示？numpy数据存储在哪里，如何显示(没有numpy包)？
示例代码：
import torch
a = torch.tensor([[1,2,3],[4,5,6]])
a = a.numpy()
print(a)
array([[1, 2, 3],
[4, 5, 6]])
浏览 0提问于2020-06-09得票数 0
1回答
我正在使用PyTorch代码在无监督setting.However中训练自定义损失函数，在训练阶段，损失不会下降，并且在可能的时期保持不变。请参阅下面的训练代码片段：
X = np.load(<data path>) #Load dataset which is a numpy array of N points with some dimension each.
num_samples, num_features = X.shape
gmm = GaussianMixture(n_components=num_classes, covariance_type='sphe
浏览 0提问于2018-02-25得票数 0
在Ubuntu20.04上，我无法导入protobuf：
(r-torch) brucezepplin@brucezepplin-HP-ProBook-440-G7:~$ conda list | grep protobuf
libprotobuf 3.17.2 h4ff587b_1
protobuf 3.17.2 py39h295c915_0
(r-torch) brucezepplin@brucezepplin-HP-ProBook-440-G7:~$ python
Pyt
浏览 6提问于2021-12-13得票数 0
回答已采纳
我是pytorch的新手，我正在尝试获得1*4096格式的预训练模型VGG16特征向量的输出，该输出是由最后一层之前的层返回的。我发现keras中也有类似的特性。在pytorch中有没有什么直接的命令来做同样的事情？
我使用的代码如下：
import torch
from torch import nn
from torch import optim
import torch.nn.functional as F
from torchvision import models
from torch.autograd import Variable
from PIL import Image
i
浏览 38提问于2019-11-23得票数 0
回答已采纳
2回答
预训练模型的预加载
、、、、
我用以下代码保存了我的模型：
from google.colab import files
torch.save(net, 'model.pth')
# download checkpoint file
files.download('model.pth')
然后以这种方式上传并检查图像(x)：
model = torch.load('model.pth')
model.eval()
torch.argmax(model(x))
在以前的会话中，它工作得很好，但是后来我启动了一个新会话，并尝试了上面的代码，并得到了这样一个错误：
Attribu
浏览 0提问于2021-05-02得票数 0
回答已采纳
require 'torch';
require 'nn';
require 'nnx';
mnist = require 'mnist';
fullset = mnist.traindataset()
testset = mnist.testdataset()
trainset = {
size = 50000,
data = fullset.data[{{1,50000}}]:double(),
label = fullset.label[{{1,50000}}]
}
validationset =
浏览 0提问于2016-05-22得票数 1
1回答
计算两点之间的直接欧几里德距离相当简单：
import torch
p1 = torch.tensor([1.0, 3.5])
p2 = torch.tensor([5.0, 9.2])
dis = torch.sum(torch.square(p1-p2))
dis
>>> tensor(48.4900)
但是，如果不经过圆，如何计算圆上两点之间的距离？也就是说，圆周上的距离，在D维空间中。
显然，在2D中，圆只是一个圆：
import numpy as np
import matplotlib.pyplot as plt
def circle_points(r, n)
浏览 19提问于2021-05-20得票数 3
回答已采纳
我看到这个：
./install/share/lua/5.1/image/init.lua: return torch.Tensor().libjpeg.size(filename)
张量是如何得到力场的？我没有看到它在Tensor.lua中被引用。
关于更多上下文，我试图调试此错误：
/Users/un/torch/install/share/lua/5.1/image/init.lua:140: attempt to index field 'libjpeg' (a nil value)
stack traceback:
/Users/un/torch/ins
浏览 1提问于2015-02-14得票数 1
回答已采纳
我想要为我的代码定义一个在pytorch中的优化器。
该代码需要有一个负学习率的一个特殊的乳胶。
当我编写这样的代码时：
import torch
class Net(torch.nn.Module):
def __init__(self):
super().__init__()
self.linear1 = torch.nn.Linear(3, 2)
def forward(self, x):
return self.linear1(x)
if __name__ == '__main__':
net
浏览 1提问于2022-07-28得票数 1
1回答
DiffSharp中不可用的函数
、、、、
我已经使用以下DiffSharp脚本成功安装了TorchSharp (以及TorchSharp、Libsharp等)：
#r "nuget: DiffSharp.Core"
#r "nuget: DiffSharp.Backends.Reference"
#r "nuget: DiffSharp.Data"
#r "nuget: DiffSharp.Backends.Torch"
#r "nuget: TorchSharp"
#r "nuget: LibTorchSharp"
open Syste
浏览 14提问于2022-10-27得票数 0
我尝试使用我的GPU来运行YOLOR模型，但我一直收到错误： Traceback (most recent call last):
File "D:\yolor\detect.py", line 198, in <module>
detect()
File "D:\yolor\detect.py", line 41, in detect
device = select_device(opt.device)
File "D:\yolor\utils\torch_utils.py", line 47, in
浏览 2818提问于2021-07-28得票数 0
1回答
我在我的主文件夹中使用了来自/usr/share/lua/5.1的包，我有一个跑步火炬版本。然后我重新安装了火炬(删除了家里的文件夹，得到了一个新的文件夹，并运行了“开始使用火炬”的安装脚本)。所有的包都在/usr/share中，但是当运行新的火炬时，我会得到它们丢失的错误。例如，软件包cv：
/home/nbackfisch/torch/install/bin/luajit:未找到/home/nbackfisch/torch/install/share/lua/5.1/trepl/init.lua:389: /home/nbackfisch/torch/install/share/lua/
浏览 1提问于2017-04-03得票数 0
回答已采纳
我试图在torch7中计算两个直方图之间的距离，为了做到这一点，我在考虑使用地球推土机的距离。现在，我知道在python中使用像这样的工具并不难，但是我在torch中有我的数据，需要多次执行这个计算。因此，在torch7和python之间移动整个数据不是一个选项。
那么，我的问题是，在torch7中最快的地球移动距离计算器是什么？我已经搜索过任何类似于库的东西，并且希望有更好的方法来实现这一点，然后逐行翻译python代码，特别是看到torch如何更好地处理gpu上的事情。
编辑我已经找到，但不知道如何使用它。
我目前有以下代码：
function ColourCompareHistE
浏览 1提问于2017-07-02得票数 2
回答已采纳
"torch.utils.data.TensorDataset“和"torch.utils.data.Dataset”之间的区别是什么--文档对此并不清楚，我在谷歌上也找不到任何答案。
浏览 173提问于2021-05-25得票数 1
回答已采纳
我试图在google中使用这个包，但它不起作用。我在文章的最后一行中添加了代码的图像。这是我的密码-
import torch
from torch.autograd import Variable
import cv2
!git clone https://github.com/amdegroot/ssd.pytorch.git #cloning the package in colab first
!pip install -q BaseTransform
from data import BaseTransform, VOC_CLASSES as labelmap
每当我试图安装Ba
浏览 2提问于2019-03-30得票数 0
我试图运行以下代码：
l = torch.tensor([0, 1, 1, 1], requires_grad=False)
r = torch.rand(4, 2)
torch.nn.CrossEntropyLoss(r, l)
我的错误是：
RuntimeError: Boolean value of Tensor with more than one value is ambiguous
我在这里看了看：，但不明白答案。
为了运行代码，我需要更改什么？
浏览 137提问于2022-06-21得票数 1
回答已采纳
我正在尝试在源代码中追踪torch.nn.NLLLoss的实现。我在文件torch.nn.functional中的函数nll_loss中调用了torch._C.nll_loss。但是我找不到创建_C的地方。
有谁有这方面的消息吗？
浏览 3提问于2018-02-20得票数 8
3回答
导入PyTorch时出错
、、、
我尝试在终端中使用以下命令安装PyTorch：
pip install http://download.pytorch.org/whl/torch-0.2.0.post1-cp27-none-macosx_10_7_x86_64.whl
然后在python中运行以下代码：
import torch
torch.__file__
并得到以下错误：
File "/Users/brian/anaconda/lib/python2.7/site-packages/torch/__init__.py", line 48, in <module>
import torch.
浏览 4提问于2017-08-14得票数 1
回答已采纳
我用pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101在conda环境中安装了pytorch和torchvision，据我理解，这意味着pytorch库是用cuda10.1编译的。在运行nvcc --version时，我得到
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V
浏览 10提问于2021-12-25得票数 1
回答已采纳
我正在尝试重写一个tensorflow脚本。我在torch中为的下一行找出了一个等价部分。
import tensorflow_probability as tfp
tfd = tfp.distributions
a_distribution = tfd.TransformedDistribution(
distribution=tfd.Normal(loc=0.0, scale=1.0),
bijector=tfp.bijectors.Chain([
tfp.bijectors.AffineScalar(shift=self._mean
浏览 15提问于2022-07-25得票数 2
回答已采纳
2回答
我有一个模型，我训练基于火炬在GPU。现在我想把它上传到cpu上。我正在使用这段代码来保存模型的加载。
以下是我的模型和培训阶段：
model = VAE(input_size , lead, hidden_dim,hidden_dim1,hidden_dimd, latent_dim, device, num_hidden= lead).to(device)
optimizer = torch.optim.Adam(model.parameters(), lr =learning_rate )
losses, kl_loss, l
浏览 18提问于2022-09-24得票数 0
回答已采纳
1回答
R网状变压器库找不到火炬
、、、
使用R和reticulate软件包，我正在尝试使用一个来自Huggingface的预先训练的模型。这种局部模型需要PyTorch和变压器。两者都可以通过网络在R中获得，但是即使我可以安装和加载两者，变压器包也找不到PyTorch安装。
use_virtualenv("r-reticulate")
reticulate::py_install('transformers', pip = TRUE)
reticulate::py_install("PyTorch")
transformer = reticulate::import('tra
浏览 41提问于2021-12-07得票数 0
无法安装最新版本的Torch (pip安装torch==1.0.1.post2) pip安装torch==1.0.1.post2 尝试在Python版本3.7和windows 10中执行操作 C:\Users\Admin>pip install torch==1.0.1.post2 Collecting torch==1.0.1.post2找不到满足要求的版本torch==1.0.1.post2 (来自版本: 0.1.2、0.1.2.post1)找不到与torch==1.0.1.post2匹配的分发版本
浏览 435提问于2019-03-24得票数 4
2回答
我正在尝试实现一个连体网络与两个图像之间的排名损失。如果我定义了我自己的损失，我能像下面这样做反向传播步骤吗？当我运行它时，有时在我看来，它给出的结果与单一网络给出的结果相同。
with torch.set_grad_enabled(phase == 'train'):
outputs1 = model(inputs1)
outputs2 = model(inputs2)
preds1 = outputs1;
preds2 = outputs2;
alpha = 0.02;
w_r = torch.tensor(1).cuda(
浏览 4提问于2018-08-28得票数 0
1回答
我的代码：
import torch
import torch.nn as nn
import torch.nn.functional as F
class MultivariateLinearRegressionModel(nn.Module):
def __init__(self):
super().__init__()
self.linear = nn.Linear(3,1)
def forward(self,x):
# print(1)
return self.linear(x)
x_train = torch.FloatTensor([[73
浏览 3提问于2021-06-27得票数 0
回答已采纳
我使用我的视频数据集训练并保存了一个模型，用于使用git-集线器( seq2seq -)的代码进行手语识别。
我创建了一个pyTorch模型来对图像进行分类。我曾经通过state_dict保存过它，整个模型是这样的：
torch.save(model.state_dict(), "model1_statedict")
torch.save(model, "model1_complete")
我装好了我的模型：
model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH)
浏览 3提问于2022-06-29得票数 0
因此，我需要访问安装ImageNet的远程机器。我已经在本地计算机上下载了训练神经网络所需的tar，我使用random_split()将每个tar拆分为80%的训练集、10%的valset和10%的testset。每组图像的名称都被附加到txt文件中，我希望使用该文件作为检查通道。
所以，我编写了python脚本，我想在远程机器上启动它。该文件必须经过ImageNet数据库的文件夹，并且基于我发送到机器的txt文件，只包括文件中“命名”的图片。下面是剧本：
import os
import torch
from torch.utils.data import random_split
from
浏览 13提问于2022-03-04得票数 1
下面是用于PyTorch 1.0.1的代码
import torch
import torch.utils
import torch.multiprocessing as multiprocessing
from torch.utils.data import DataLoader
from torch.utils.data import SequentialSampler
from torch.utils.data import RandomSampler
from torch.utils.data import BatchSampler
from torch.utils.data impo
浏览 19提问于2022-01-28得票数 2
问题：我如何在google中下载PyTorch几何的旧版本？
Context：我正在尝试使用/加载一个吡传几何图形，并得到错误消息："RuntimeError：'data‘对象是由一个旧版本的PyG创建的。如果这个错误发生在加载一个已经存在的数据集时，请删除dataset根文件夹中的'processed/’目录，然后再试一次。“这张图是在2021年夏天生成的。
我正在使用PyTorch和下面的代码来导入Pythor几何学，但是在使用旧版本时仍然会出现错误。
#import torch
!pip install torch==1.8.0
import torch
torc
浏览 6提问于2021-12-12得票数 1
我有一个非常简单的场景。我有一个自定义类(而不是NN模型)，我正在使用torch.save进行腌制。当我试图加载它时，它会出现错误：
Traceback (most recent call last):
File "/Applications/PyCharm.app/Contents/plugins/python/helpers/pydev/_pydevd_bundle/pydevd_exec2.py", line 3, in Exec
exec(exp, global_vars, local_vars)
File "<input>"
浏览 4提问于2021-03-17得票数 0
1回答
我试图修改torch中R中张量的梯度
library(torch)
x <- torch_randn(10,2)
x$grad <- torch_randn(10,2)
但是，我得到以下错误：
Erreur dans f(value) : argument inutilisé (value)
2: `$<-.R7`(`*tmp*`, grad, value = <pointer: 0x55cd40aefae0>)
1: `$<-`(`*tmp*`, grad, value = <pointer: 0x55cd40aefae0>)
我以为我做错了
浏览 3提问于2022-04-27得票数 2
2回答
我正在学习关于聊天机器人的教程：
我有窗户10。
import torch #works
但下列进口产品不起作用：
import torch.nn
import torch.jit
错误：
没有找到名为torch.nn的模块
和
没有找到名为torch.jit的模块
我在蟒蛇中安装了下列设备：
conda install pytorch-cpu torchvision-cpu -c pytorch
我应该怎么做才能删除错误？
守则是：
import torch
from torch.jit import script, trace
import torch.nn as nn
from
浏览 1提问于2019-04-04得票数 0
如果我有模特：
import torch
import torch.nn as nn
import torch.optim as optim
class net_x(nn.Module):
def __init__(self):
super(net_x, self).__init__()
self.fc1=nn.Linear(2, 20)
self.fc2=nn.Linear(20, 20)
self.out=nn.Linear(20, 4)
def for
浏览 5提问于2021-05-09得票数 5
回答已采纳
在torch==1.7.1+cu101中，我有两个张量
import torch
a = torch.rand(1,5,10)
b = torch.rand(100,1,10)
和一个前馈网络
import torch.nn as nn
l = nn.Linear(10,10)
我强迫其中一行是平等的：
a[0,0] = b[80][0].clone()
然后，我将两个张量都提供给l。
r1 = l(a)
r2 = l(b)
显然，由于a[0,0]等于b[80,0]，r1[0,0] 必须等于 r2[80,0]。但结果却是：
(r1[0,0] == r2[80,0]).all()
>>
浏览 2提问于2021-10-06得票数 3
回答已采纳
我有一个类和构造函数，我在__init__中添加了一个额外的参数，现在我得到了错误，TypeError: FeatureDataset() takes 1 positional argument but 2 were given。
我想知道为什么。在我看来，它应该接受两个论点。这是一个不完全函数，但我想通过这个构造函数参数编号错误。我检查了几个答案，它们要么是关于一些特别不同的东西，要么是缩进，而我没有这些问题(每一个新的缩进除以4个缩进)。
def FeatureDataset(Dataset):
def __init__(self, root_dir, file_name):
浏览 4提问于2022-05-06得票数 0
回答已采纳
我正在尝试部署我的Django应用程序，它使用机器学习模型。而机器学习模型需要pytorch来执行。当我尝试部署时，它给了我这个错误
ERROR: Could not find a version that satisfies the requirement torch==1.5.0+cpu (from -r /tmp/build_4518392d43f43bc52f067241a9661c92/requirements.txt (line 23)) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2, 0.4.1, 0.4.1.post2, 1.0
浏览 1提问于2020-05-17得票数 7
我正在尝试用LibTorch 1.3和C++创建NN，使用Cuda10.1和Windows10。对于构建，我使用VisualStudio2019。
到目前为止，我尝试了基本的例子和与CPU一起工作。但是，我不能用CUDA运行它。我试图将模型移动到GPU，因为它被描述为，但它不起作用。
要将模型移动到GPU内存，可以编写model.to(at::kCUDA)；通过调用tensor.to(at::kCUDA)，确保模型的输入也存在于CUDA内存中，这将在CUDA内存中返回一个新的张量。
所以我试了很简单
int main(){
auto net = std::make_shared
浏览 1提问于2019-11-15得票数 3
1回答
放电筒中LSTM的参数和函数调用
、、、、
有谁能解释一下下面的代码：
import torch
import torch.nn as nn
input = torch.randn(5, 3, 10)
h0 = torch.randn(2, 3, 20)
c0 = torch.randn(2, 3, 20)
rnn = nn.LSTM(10,20,2)
output, (hn, cn) = rnn(input, (h0, c0))
print(input)
在调用rnn rnn(input, (h0, c0))时，我们在括号中给出了参数h0和c0。这是什么意思？如果(h0，c0)表示单个值，那么这个值是什么，这里传递的第三个参数是
浏览 1提问于2021-07-20得票数 0
回答已采纳
目前，我正在尝试加载我的预培训的GoogleNet。然而，似乎存在一个大小不匹配的问题，我试图改变num_classes来解决它，但没有效果，仍然存在问题。
import os
from tkinter import Variable
from matplotlib import image, transforms
import torch
import torchvision
from torch import nn, optim
checkpoint = torch.load("extraFile/Kaggle_googlenet.pth")
model = torchv
浏览 14提问于2022-05-02得票数 0
我试图从default_collate中导入torch.utils.data.dataloader，但是它给了我一个default_collate不存在并且存在另一个函数(_collate_fn_t)
从torch.utils.data.dataloader导入default_collate从torch.utils.data.dataloader导入_collate_fn_t
这些命令是否相似？
浏览 7提问于2020-06-25得票数 1
回答已采纳
在PyTorch中有两个张量，z是形状(n_samples, n_features, n_views)的三维张量，其中n_samples是数据集中的样本数，n_features是每个样本的特征数，n_views是描述相同(n_samples, n_features)特征矩阵的不同视图数，但与其他值相同。
我有另一个二维张量b，形状为(n_samples, n_views)，它的目的是在不同的视图中重新描述样本的所有特征。换句话说，它封装了同一个示例每个视图的特性的重要性。例如：
import torch
z = torch.Tensor(([[2,3], [1,1], [4,5]],
浏览 0提问于2021-06-07得票数 2
回答已采纳
1回答
我正在使用服装模型进行分割(SETRModel)。模型输出形状是(nBatch，256,256)，下面的代码确认了这一点(请注意，通道已被挤出)。目标形状是相同的(它是一个PILMask)。 当我开始训练时，我得到一个与损失函数相关的运行时错误(见下文)。我做错了什么？ 大小= 480 half= (256,256) 拆分器=FuncSplitter(λo: Path(o).parent.name == 'validation') dblock = DataBlock(blocks=(ImageBlock，MaskBlock(代码)， get_item
浏览 42提问于2021-04-26得票数 0
回答已采纳
当我将PyTorch用于C++时，很容易使用find_package来设置依赖关系。这是CMakeLists.txt
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(dcgan)
set(CMAKE_PREFIX_PATH /User/root/libtorch) # I added this line, does it effect?
find_package(Torch REQUIRED)
add_executable(dcgan dcgan.cpp)
target_link_libraries(dcgan "${
浏览 1提问于2020-07-09得票数 1
回答已采纳
1回答
我在玩一些代码，它的行为与我预期的不一样。所以我把它简化为一个最简单的例子：
import torch
test_act = torch.tensor([[2.,0.]])
test_target = torch.tensor([0])
loss_function_test = torch.nn.CrossEntropyLoss()
loss_test = loss_function_test(test_act, test_target)
print(loss_test)
> tensor(0.1269)
weights=torch.tensor([0.1,0.5])
loss_f
浏览 2提问于2021-05-21得票数 0
回答已采纳
在多GPU机上使用PyTorch估计器和SageMaker对图形神经网络进行训练/微调。
安装到Estimator容器中的requirements.txt的行如下：
torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
torch-cluster -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
torch-spline-conv -f
浏览 12提问于2022-03-31得票数 0
回答已采纳
目前，我正试图通过遵循将自定义模型部署到AI平台上。它是基于'torchvision.transform'.和‘Py手电筒’的预训练模型的组合。目前，我一直处于与500 on的自定义预测约束有关的错误下面。
错误：(gcloud.beta.ai-platform.versions.create)创建版本失败。错误检测模型:模型需要比允许的内存更多的内存。请尽量缩小型号尺寸并重新部署。如果您继续遇到错误，请与支持部门联系。
Setup.py
from setuptools import setup
from pathlib import Path
base = Path(__f
浏览 0提问于2021-01-19得票数 0
回答已采纳
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
云服务器
ICP备案
对象存储
即时通信 IM
实时音视频活动推荐
腾讯云开发者
