如何在PyTorch中从预先训练好的模型中获取权重并在TensorFlow中使用？这是预先训练好的模型：
lstm = torch.hub.load("BruceWen120/medal", "lstm")
浏览 3提问于2021-11-28得票数 0
我可以让它在CPU上工作，但我更喜欢在GPU上运行它。取而代之的是，我得到了如下错误： RuntimeError:应为后端CPU的对象，但获得了参数#2 'weight‘的后端CUDA RuntimeError:应为后端CPU的对象，但获得了参数#4 'mat1‘的后端CUDA 所以，就像我的帖子问的那样，我如何才能让预先训练好的模型<
浏览 34提问于2020-06-20得票数 0
回答已采纳
1回答
我有一个特定的计算机视觉问题，我想尝试使用一些预先训练的模型来解决。与其他ML框架(例如PyTorch)不同，Flux.jl文档实际上没有任何预先训练好的模型。我如何在Flux中访问这些相关的模型？
浏览 22提问于2021-07-11得票数 0
回答已采纳
关于CoreML模型的教程都是将TensorFlow模型或PyTorch模型转换为CoreML模型。如果预先训练好的模型文件是mlmodel类型，我不知道如何在python中使用它，也不知道如何将mlmodel转换为tf模型。如何在Python语言中通过TensorFlow、Keras或PyTorch来读取和部署CoreML模型
浏览 28提问于2020-09-06得票数 0
回答已采纳
我正在使用Pytorch进行图像分类。我正在寻找在ImageNet以外的数据集上预先训练的CNN模型，我找到了一个指向".ckpt“文件的链接。我还找到了使用Tenserflow加载此文件的教程，但没有使用pytorch。如何从".ckpt“文件中使用Pytorch加载预先训练好的模型？
浏览 295提问于2019-11-25得票数 1
回答已采纳
我的GPU是NVIDIA RTX 2080 TITensorflow-gpu 1.12.0加载构建模型后(在编译之前)，我发现GPU内存已完全分配可能的原因是什么，我如何调试它？我没有多余的内存来加载数据，即使我通过生成器加载
我试着监控GPU<
浏览 9提问于2019-07-16得票数 0
1回答
我使用huggingface transformer在Pytorch中微调了一个预先训练好的BERT模型。所有的训练/验证都在云中的GPU上完成。现在，我将saved_model目录下载到我的计算机中，并希望加载模型和记号赋予器。我可以像下面这样加载模型 model = torch.load('./saved_model/pytorch_mo
浏览 19提问于2019-10-16得票数 8
回答已采纳
我可以在gpu内存中加载模型和数据样本，但是当我用示例对模型进行前向调用时，它会出现CUDA out of memory错误。我确信模型和数据已经加载，因为我的代码结构如下(伪代码)：sample = load_sample()
sleep(5) # to check memory usage我假设在构建模型的过程中为向前传递分配了所有必要的内存，但我不知道如何会发生这样的错误。我也无法在谷歌
浏览 3提问于2019-12-10得票数 0
回答已采纳
1回答
我对机器学习的整个领域都是新手，但我之前确实有过一些AWS识别的经验。在Rekognition中，您可以自定义图像中的不同部分，而不仅仅是整个图像。我希望在Tensorflow中做一些类似的事情，但尽管浏览了他们的文档并进行了搜索，但我还是找不到任何东西。 这在Tensorflow中是不可行的，还是我错过了什么？如果我在单独建模的图像上训练一个模型，它是否可以选择图像中所有这些出现的位置-例如。有5个人的图像，所有5个人都被单独检测到
浏览 24提问于2020-06-26得票数 0
回答已采纳
1回答
Pytorch中的预训练模型
、、、、
我刚刚加载了预先训练好的模型3Dresnet https://github.com/kenshohara/3D-ResNets-PyTorch。从上面的链接下载文件'resnext-101-kinetics.pth‘后，我以这种方式加载了这个模型：way i defined the model 这个模型适用于视频的大数据集(不幸的是，我的pc上没有足
浏览 25提问于2019-04-23得票数 0
我需要在c++中运行一个预先训练好的pytorch nn模型(用python训练)来进行预测。为此，我遵循这里给出的关于如何在c++中加载pytorch模型的说明：https://pytorch.org/tutorials/advanced/cpp_export.html 但是，当我尝试通过本教程第一步中所述的跟踪
浏览 36提问于2019-02-12得票数 3
回答已采纳
我正尝试在亚马逊网络服务ec2上运行StackGAN代码，但演示程序不起作用。我已经下载了文本嵌入并对数据进行了预处理。然而，我一直收到这个错误我尝试手动安装pytorch，但在ec2上构建失败。我还进入了pytorch_p27提供的torch环境，但出现了相同的错误。
我
浏览 1提问于2018-03-25得票数 0
我有一个现有的模型，在这个模型中，我加载了一些预先训练好的权重，然后在pytorch中进行预测(一次一个图像)。我正在尝试将它基本上转换为pytorch闪电模块，并对一些事情感到困惑。因此，目前，我的模型__init__方法如下所示： self.lightning如何知道它是否需要使用GPU？我猜这需要在某个地方指定。在某种程度上，我想扩展这个模型实现来做训练，所以我想确保我做<e
浏览 263提问于2020-05-03得票数 3
回答已采纳
3回答
在运行ubuntu 16.04中的脚本时，我得到了这个错误。请原谅我，我是python的新手，我已经检查了互联网上已经可用的选项，但我不能修复它。RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module.cpp:32
我当前正在运行此文件model.load_state_dict(torch.load(options["general"]
浏览 0提问于2018-06-10得票数 7
回答已采纳
我按照https://modelzoo.co/model/pytorch-cyclegan-and-pix2pix中给出的步骤下载了一个预先训练好的模型。这里给出的前3个命令如下： git clone https://github.com/pytorch/vision**python setup.py install** 然而，当我运行</em
浏览 19提问于2020-01-15得票数 0
回答已采纳
2回答
目前，我正在处理一个数据集，其中我有两种图像：“锐化版本”的图像和“模糊版本”的相同图像，其中模糊是人工添加的。我的目标是训练一个模型，该模型接受模糊版本的图像，并尝试尽可能多地对图像进行去模糊处理，以便“去模糊图像”更接近于锐化版本。在文献中，UNet架构似乎是一个效果良好的模型。此外，我还可以通过Pytorch ()使用预先训练好的U-Net。我现在的
浏览 33提问于2021-06-02得票数 0
我正在学习Pytorch，并试图理解这个库是如何进行语义分割的。到目前为止，我所理解的是，我们可以在pytorch中使用预先训练好的模型。我找到了一篇在.eval()模式下使用该模型的文章，但我还没有找到任何关于在我们自己的数据集上使用这种模型进行训练的教程。我有一个非常小的数据集，我需要迁移学习来获得结果。我的</em
浏览 27提问于2019-12-13得票数 0
我有一个高分辨率无人机图像的数据集，我已经将其拆分成大小为512x512的瓦片。这些瓷砖将通过描绘特定类型的植物进行注释，并以COCO格式存储。我想使用自监督学习，例如在tiles中进行分割。Pytorch Lightning bolts模块提供了已经预先训练好的自我监督模型，例如SimCLR。根据bolts module documentation的说法，应该可以使用螺栓在任何计算机视觉任务上微调SimCLR。 然
浏览 33提问于2020-12-11得票数 3
1回答
并且可以方便地加载预先训练好的模型，如VGG19。但我发现使用pytorch不是这样的，因为conv2d()不接受显式的内核，而是接受内核大小。所以我想知道，我们是否可以通过简单地将权重传递给像conv2d()这样的方法来重用VGG19中的权重。任何回复都将不胜感激。
浏览 4提问于2017-09-18得票数 1
2回答
我在服务器上运行代码。那里有两个GPU，第一个很忙。然而，我找不到一种在它们之间切换的方法。如果这很重要，我会使用pytorch。
浏览 26提问于2021-02-17得票数 0
回答已采纳
云服务器
ICP备案
对象存储
腾讯会议
云直播
腾讯云开发者
