torch 相关问题

Torch是LuaJIT的科学计算框架。它被世界各地的机器学习研究人员广泛使用。

KeyError:'optimizer_state_dict'

我尝试在另一个数据集上使用预训练的模型,以使用相同的代码在我的数据集上恢复训练,但是下载模型并运行 main.py 到 laod 模型后出现错误 这个

回答 1 投票 0

nn.BCELoss() 给出的结果非常糟糕

我正在尝试使用表格数据集构建一个 PyTorch 分类程序,我的模型具有以下架构: 批次大小 = 8 纪元 = 10 隐藏神经元 = 25 LR = 1e-3 类 MyModel(nn.Modu...

回答 1 投票 0

如何在Vs代码中设置flask

您的机器中必须安装有 python。 创建Python虚拟环境并激活虚拟环境链接 使用以下命令安装所有依赖项 pip install -r 要求...

回答 1 投票 0

Pytorch CustomDataset TypeError:列表索引必须是整数或切片,而不是列表

我在使用 pytorch 自定义数据集时遇到错误。这个问题对我来说真的很奇怪,因为它正在工作,我没有更改代码上的任何内容。这是场景: 在建立了我的深度学习之后...

回答 1 投票 0

如何对 PyTorch 张量进行哈希计算

给定一个整数的 PyTorch 张量,例如 torch.Size(N, C, H, W)。 有没有一种方法可以有效地散列张量的每个元素,以便我获得从 [-MAX_INT32 到 +MAX_INT32] 或 [...

回答 1 投票 0

运行时错误:在您的系统上找不到 NVIDIA 驱动程序

我在使用 Torch 加载模型时遇到问题,该模型是使用 GPU 训练的,我正在尝试使用 CPU 加载该模型。但是我成功地加载了模型,但同时预测了

回答 1 投票 0

在生成最终图像之前获取图像迭代StableDiffusionPipeline.pretrained

我目前正在使用扩散器 StableDiffusionPipeline (来自拥抱脸)通过我和朋友一起使用的不和谐机器人生成 AI 图像。我想知道是否有可能获得 prev...

回答 1 投票 0

如何解决导入错误错误:无法从“transformers”导入名称“GenerationConfig”

我正在使用 Paperspace 进行编程。我想从 trl 导入 AutoModelForCausalLMWithValueHead、PPOConfig、PPOTrainer 但遇到错误。我安装了所需的库,例如...

回答 1 投票 0

使用 PyTorch 教程微调 Mask R-CNN 的问题:Torch 和 Torchvision 的模块错误

我一直在遵循这个 PyTorch 教程,使用我自己的数据集微调 Mask R-CNN 模型:PyTorch 教程 (https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html)。在纸面上,它...

回答 1 投票 0

如何在slurm中设置MASTER_PORT和MASTER_ADDR

在torch官方文档中谈到DDP时,说要设置如下: def 设置(等级,世界大小): os.environ['MASTER_ADDR'] = 'localhost' os.environ['MASTER_PORT'] = ...

回答 1 投票 0

`torch.einsum` API 是如何工作的?

torch.einsum API 如何工作? 我试图理解如何 torch.einsum("ac,bc->ab",norm_max_func_embedding,norm_nl_embedding) 正在计算相似度? 我明白这是做...

回答 1 投票 0

argmax() 函数是否有可微的近似值

输出一个向量,对应输入向量中最大值的位置接近1,其他位置接近0。这个函数需要可微。 例如: ...

回答 1 投票 0

Pytorch batchnorm2d:“运行时错误:running_mean 应包含 1 个元素而不是 64”

类似的问题都没有解决。所以请不要标记为重复 Pytorch BatchNorm2d 需要 N C H W 格式的输入 在哪里 N = 批量大小 C = 通道 H = 高度 W = 宽度 正如他们在...

回答 1 投票 0

如何在虚拟环境中的 Windows 上使用 pip 和 python 3.8.3 安装 torch 和 torchvision?

我为我的项目创建了一个虚拟环境,我正在尝试通过 pip 安装 torch。 pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html 这给...

回答 1 投票 0

使用 CUDA 张量和 CPU 张量的 PyTorch 前向传递,同时保留梯度

我有一个由神经模型产生的向量a,它需要与一个巨大的矩阵M交互。由于M很大,我必须在CPU设备中进行计算。这样的话,我想知道渐变是否可以

回答 1 投票 0

Torch:类型错误:无法腌制“WeakMethod”对象

我想使用 torch.save() 保存权重。 我遇到了这个问题 -------------------------------------------------- 类型错误:无法腌制“WeakMethod”对象 ----------------------------...

回答 1 投票 0

有谁知道 TypeError: Cannot pickle 'WeakMethod' object

我想使用 torch.save() 保存权重。 我遇到了这个问题 类型错误:无法腌制“WeakMethod”对象 我的代码是这样的 文件 home/4DenoiseNet/networks/train/tasks/semantic/../../...

回答 1 投票 0

Databricks AWS 内存差异

我正在使用 Databricks 在 Pytorch 中训练/测试模型,但我不断遇到没有意义的内存错误。在 Databricks 作业的配置中,我指定了 node_type_id 和

回答 1 投票 0

如何保存和加载 Peft/LoRA Finetune(明星聊天)

我正在尝试进一步微调Starchat-Beta,保存我的进度,加载我的进度,并继续训练。但无论我做什么,它都不会在一起。每当我加载进度并继续训练时...

回答 1 投票 0

Pytorch 简单线性回归模型:模型参数未更新

我是 PyTorch 的新手。我一直在关注 PyTorch 创建简单线性回归模型的教程之一。 我已经相应地跟进了这里的代码供参考。 LinearRegr 类...

回答 1 投票 0

© www.soinside.com 2019 - 2024. All rights reserved.