pytorch 相关问题

PyTorch是一个深度学习框架,它实现了一个动态计算图,它允许您改变神经网络在运行中的行为方式,并能够执行向后自动区分。

PyTorch DataLoader 和 Matplotlib 的 Imshow 之间的图像分类任务问题

我目前正在研究涉及图像数据的二元分类任务。首先,我必须检查我的数据集。但是,我遇到了 DataLoader 的问题。 在

回答 1 投票 0

在 Docker 容器中使用 Python 版本 3.11 安装 PyTorch

我在PyTorch官方页面上看到PyTorch支持Python版本3.8到3.11。 当我实际尝试在 Python 3.11 Docker 映像中安装 PyTorch + CUDA 时,似乎无法找到 CUDA 驱动器...

回答 1 投票 0

断言错误:Torch 未在启用 CUDA 的情况下编译

来自 https://pytorch.org/ 要在 MacOS 上安装 pytorch,请执行以下操作: conda 安装 pytorch torchvision -c pytorch # MacOS 二进制文件不支持 CUDA,如果需要 CUDA,请从源代码安装...

回答 3 投票 0

将包含图像的张量展开为补丁

我有一批尺寸为 4、尺寸为 h x w = 180 x 320 的单通道图像。我想展开它们一系列 p 个形状为 h_p x w_p 的小块,产生形状为 4 x p x h_p x w_p 的张量。如果 h 不是

回答 1 投票 0

如何删除GPU内存中的变压器模型

从变压器导入 CTRLTokenizer, TFCTRLLMHeadModel tokenizer_ctrl = CTRLTokenizer.from_pretrained('ctrl',cache_dir='./cache',local_files_only=True) model_ctrl = TFCTRLLMHeadModel.from_pret...

回答 2 投票 0

错误:无法为 mmcv-full 构建轮子,这是安装基于 pyproject.toml 的项目所必需的

python = 3.10 火炬=2.2.1+cu118 pip 安装 mmcv-full 运行 install_scripts C:\Users\omras\miniconda3 nvs ransseg1\lib\site-packages\wheel dist_wheel.py:109: RuntimeWarning: 配置

回答 1 投票 0

如何使用加速在连续的 PyTorch 训练阶段之间正确管理 GPU 内存?

我遇到了一个具有挑战性的问题,GPU 内存在 PyTorch 的连续训练阶段之间没有正确释放,导致 CUDA 内存不足错误。 我的项目涉及微调

回答 1 投票 0

如何恢复 BERT/XLNet 嵌入?

我最近一直在尝试堆叠语言模型,并注意到一些有趣的事情:BERT 和 XLNet 的输出嵌入与输入嵌入不同。比如说这个...

回答 2 投票 0

创建 pyTorch 测试数据集(无标签)

我为我的训练数据创建了一个 pyTorch 数据集,其中包含特征和标签,以便能够使用本教程使用 pyTorch DataLoader。这对于我的训练数据来说效果很好,...

回答 2 投票 0

在torchrl中为pettingzooenv定义rewardsum实例

我想在torchrl中为pettingzoo env包装器定义rewardsum()实例。 这是我的环境的定义: 从 torchrl.envs.libs.pettingzoo 导入 PettingZooEnv 从 torchrl.envs.utils 导入

回答 1 投票 0

PyTorch LSTM 模型未训练

我正在尝试训练一个 LSTM 模型,然后训练一个全连接层,以对一组具有 22 个通道(序列长度为 1000)的 EEG 时间序列数据进行分类。我正在使用 PyTorch...

回答 1 投票 0

如何在Python中可视化torch_geometric图?

让我们考虑一个例子,我有以下坐标格式的邻接矩阵: > edge_index.numpy() = array([[ 0, 1, 0, 3, 2], [ 1...

回答 2 投票 0

为什么我在huggingface MT5模型中执行批量编码时得到不同的嵌入?

我正在尝试使用 HuggingFace 的 mt5 基础模型对一些文本进行编码。我正在使用如下所示的模型 从变压器导入 MT5EncoderModel、AutoTokenizer 模型 = MT5EncoderModel.from_pretrai...

回答 1 投票 0

如何解决pytorch中剪枝模型的深度复制错误

我正在尝试构建一个强化学习模型,其中我的演员网络有一些修剪过的连接。 当使用 torchrl 中的数据收集器 SyncDataCollector 时,深度复制失败(请参见下面的错误)。 这似乎...

回答 1 投票 0

使用火炬时多处理速度变慢

我正在使用 scipy 中的 Differential_evolution 和工人来并行计算。我从 numpy 切换到 pytorch 以加快代码速度。 从 torch.multiprocessing 导入 set_start_method,P...

回答 1 投票 0

在没有retain_graph=True的情况下反向传播两个具有不同损失的网络?

我有两个依次执行昂贵计算的网络。 两者的损失目标是相同的,除了第二个网络的损失我想应用掩码。 如何实现这一目标...

回答 1 投票 0

这段代码运行完美,但我想知道 my_forward 函数中的参数“x”指的是什么

参考VIT转换器示例中的注意图:https://github.com/huggingface/pytorch-image-models/discussions/1232?sort=old 这段代码运行完美,但我想知道参数是什么......

回答 1 投票 0

将视频保存到Torch张量

我有一个分辨率为 480x854 的长视频(~20.000 帧)。我想将整个视频保存到 torch.tensor (可能是 .pt 文件)。但是,我的 RAM 不足以一次加载整个视频。难道是……

回答 1 投票 0

AttributeError:模块“torchtext.data”没有属性“Field”未用torchtext.legacy修复

我正在尝试使用 torchtext 中的 torchtext.data.Field() 但它无法以我尝试过的任何方式工作。 我尝试过 从 torchtext.legacy 导入数据但它似乎不起作用。我也有...

回答 1 投票 0

Python 中的多个 ONNX 输出(获取 ONNX 的中间层输出)

如何将模型导出到 ONNX,以便获得中间层的输出以及层? (我见过一个类似的问题,但这里没有答案) 假设我有一个模型,模型。型号...

回答 1 投票 0

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