pytorch
文章平均质量分 69
ting♡
无脑学编程,记录,学习
展开
-
【环境搭建】pytorch
运行并选择安装路径,等待安装完成。(勾选 Add Anaconda to the system PATH environment variable,将Anaconda添加到环境变量中)查看是否安装成功,cmd中输入conda回车,是否出现如下信息,有则说明安装成功。。原创 2023-06-24 16:08:04 · 172 阅读 · 0 评论 -
【pytorch】squeeze()和unsqueeze()函数介绍
在pytorch中,我们对张量Tensor的维度进行压缩或者扩充(被压缩或者扩充的维度为1),经常使用的是squeeze()函数和unsqueeze()函数。原创 2023-06-14 11:18:26 · 364 阅读 · 0 评论 -
【bug】torch.nn.Softmax()报错.UserWarning: Implicit dimension choice for softmax has been deprecated
翻译一下,就是说用户警告:已弃用softmax的隐式尺寸选择。将调用更改为包含dim=X作为参数。torch.nn.Softmax(dim=1)可参考这篇文章,有图,挺清晰。,即在第1维上概率sum=1。原创 2023-04-18 11:56:14 · 399 阅读 · 0 评论 -
【pytorch】实现简单的CNN卷积神经网络
如果设为True,它会把输出直接覆盖到输入中,这样可以节省内存。之所以可以覆盖是因为在计算ReLU的反向传播时,只需根据输出就能够推算出反向传播的梯度。一般不使用inplace操作。在Pytorch的nn模块中,封装了nn.Conv2d()类作为二维卷积的实现。nn.ReLU()用来实现Relu函数,实现非线性。最后这个完整代码是刚学的时候网上找的,不太记得出处了。前三个参数需要手动提供,后面的都有默认值。原创 2023-04-18 11:41:55 · 1319 阅读 · 0 评论 -
【pytorch】torchsummary打印模型结构和参数信息
torchsummary 可以查看网络的顺序结构,显示每一层的类型、out shape和参数量; 还有网络参数量,网络模型大小; fp/bp 一次需要的内存大小等信息。原创 2023-04-17 18:18:32 · 2970 阅读 · 0 评论 -
【Pytorch】构建VOC2012数据集代码详解
本文使用 PASCAL VOC 数据集原创 2023-03-22 16:34:12 · 1274 阅读 · 0 评论 -
【完整代码】FCN简单实现
【代码】【完整代码】FCN简单实现。原创 2023-03-31 17:22:46 · 1347 阅读 · 3 评论 -
【pytorch】使用torchvision进行语义分割
使用已经在COCO Train 2017数据集的子集上进行训练的FCN进行语义分割,该子集对应于PASCALVOC数据集。模型共支持20个类别。原创 2022-10-11 17:51:28 · 1366 阅读 · 0 评论 -
【bug】TypeError:can’t convert cuda:0 device type tensor to numpy.
numpy不能直接读取CUDA tensor,需要将它转化为 CPU tensor。如果想把CUDA tensor格式的数据改成numpy,需要先将其转换成cpu float-tensor之后再转到numpy格式。在CPU上是正常运行的,然后用GPU的时候就出现了这个报错。会出现新的报错,记得把括号加上!他已经告诉我们修改方法了,要先把。改成这个样子就好了!原创 2023-04-10 10:53:49 · 510 阅读 · 0 评论 -
【warning】UserWarning: The parameter ‘pretrained‘ is deprecated since 0.13 and may be removed
就是参数列表中的pretrained在新版本中被弃了,要使用weights这个参数。然后教你用新的参数。就按照watning里写的把models.resnet101()后面的内容重新设置就好。原创 2022-11-19 19:24:06 · 7967 阅读 · 3 评论 -
【bug】 IndexError: Target 255 is out of bounds.
输出与标签数不一致,导致在计算loss时报错Target N is out of bounds。原创 2022-11-04 16:56:45 · 1963 阅读 · 6 评论