与保存的模型相比，加载的 PyTorch 模型具有不同的结果
我有一个 python 脚本，用于训练然后测试 CNN 模型。模型权重/参数在测试后通过以下方式保存： checkpoint = {'state_dict': model.state_dict(),'opt…
bert 输出到 cnn 模型
我正在尝试使用 pytorch 将 bert 模型与 Cnn 1d 连接起来。我使用了这段代码，但我不明白函数 conv1d 中 in_channels 和 out_channels 的含义是什么 …
如何获取pytorch脚本模型中运算符的输入大小？
我使用此代码将模型传输到脚本模型： scripted_model = torch.jit.trace( detector.model, images).eval() 然后打印 scripted_model。部分输出如下： …
您需要将 EOS 和 BOS 代币放入自动编码器变压器中吗？
我开始了解变压器架构，但有些事情我还无法掌握。 在无解码器的转换器中，例如 BERT，分词器始终包含句子前后的标记 CLS 和 SEP。我知道 CLS 既充当 B…
Pytorch 错误：TypeError：adaptive_avg_pool3d()：参数“output_size”； （位置 2）必须是整数元组，而不是列表
尝试在 Pytorch 中使用 AdaptiveAvgPool3D 时出现此错误。以下是错误跟踪 Traceback (most recent call last): 文件“/scratch/a.bip5/BraTS 2021/./s…
有没有办法在 torch 中训练具有不可训练参数的神经网络？
所以，我正在尝试学习迁移学习并尝试在自动编码器中实现它，同时我冻结了一些初始层，然后我也在其中添加了新层，现在我正在尝试训练它，但由于某些原…
加载 pytorch 检查点时出现问题 -- 存储错误大小错误 -- 如何修复？
我收到以下错误： Traceback (most recent call last): File "/home/miranda9/diversity-for-predictive-success-of-meta-learning/div_src/diversity…
FileNotFoundError：实体文件夹不存在！在谷歌 Colab 中
谁能帮我解决这个问题？ 当我在 Colab 中运行这些行时 ：param files_name: 包含训练和验证样本列表文件。 :param box_and_transcripts_folder: 包含…
如何添加“裁剪”为了连接编码器和解码器级别中的跳跃连接，如 UNET 论文中所述
我已经实现了以下UNET论文代码，这是架构： 问题在于，在第 4 级，编码器具有形状 512 x 64 x 64 的特征，而解码器部分将具有与 不同的特征形状>512 x…
只有浮点和复杂数据类型的张量才需要梯度
当我在 torch.no_grad() 上下文中运行卷积运算时，收到以下错误： 运行时错误：只有浮点和复杂数据类型的张量才需要渐变。 import torch.nn as nn imp…
Pytorch tensor.shape[3] 得到一个张量？
作为我的图片，我有一个张量：X，当我使用 SummaryWriter 到 add_graph(net, X)，X.shape[3] 变成一个 张量。为什么？它始终是一个 int。 环境：torc…
使用索引数组选择 3D pytorch 张量的第二个维度
我对 numpy 和 torch 有点陌生，我正在努力理解对我来说似乎最基本的操作。 例如，给定这个张量： A = tensor([[[6, 3, 8, 3], [1, 0, 9, 9]], [[4, 9…
无法在 YOLOR 中获得正确的 yolor 预训练权重
我正在 yolor 中训练自定义数据集。我成功地运行了一次，但一段时间后，我无法做得很好。 我注意到的第一个错误是在训练部分： Traceback (most recen…
在 pytorch 中获取较小的 MNIST 数据集
这就是我加载数据集的方式，但数据集太大。大约有 60k 张图像。所以我想将其限制为 1/10 进行训练。有什么内置方法可以做到这一点吗？ from torchvisi…
