
pytorch
文章平均质量分 74
MAUM
人工智能、数字图像处理、数字信号处理、机械故障诊断
展开
-
PyTorch出现RuntimeError: Function AddBackward0 returned an invalid gradient at index 1
感谢下面这篇文章,问题解决了:https://blog.youkuaiyun.com/andyL_05/article/details/107952479报错是这样的:RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - expected type TensorOptions(dtype=float, device=cuda:0, layout=Strided, requires_grad=false) b原创 2021-08-04 21:25:48 · 1121 阅读 · 0 评论 -
pytorch中创建多个空的tensor、pycharm从自己写的.py中引用函数下面画红线等问题
1 pytorch中创建多个空的tensor神经网络模型搭建中,如果写一些并联的模块,例如inceptionV3等这些具有多个尺度的卷积输出,其结果需要拼接起来,需要使用nn.ModuleList()使自己的代码看起来不是那么的笨拙,并且容易扩展,那么在调用这些模块的时候,需要使用enumerate()去遍历,这时候,想把并联的模块的输出分别保存下来就需要使用一tensor的list:xx = [torch.tensor([], requires_grad=False) for i in rang原创 2021-07-22 18:47:42 · 2253 阅读 · 0 评论 -
python/pytorch中的一些函数介绍
看cvt代码,记录里面的一些不认识的函数或功能。1 collections.OrderedDict包含:from collections import OrderedDict作用:建立有序的键值对集合,也就是有序字典,可以记录键值对插入的顺序,下面是我跑的代码,顺便揭露两个瞎搞的博客,误人子弟,写的就是错的!https://www.cnblogs.com/notzy/p/9312049.htmlhttps://blog.youkuaiyun.com/longshaonihaoa/article/de原创 2021-07-03 18:02:36 · 3103 阅读 · 1 评论 -
pytorch学习:运行Mask R-CNN例程
目录1 学习pytorch进行图像处理的四个必用链接:2运行Mask R-CNN例程时的问题点2.1 在哪儿下载 engine.py transforms.py utils.py?2.2 安装pycocotools2.3手动输入代码时出现的一些问题3个人对于tensorflow caffe pytorch的浅显体会1 学习pytorch进行图像处理的四个必用...原创 2020-03-04 12:36:47 · 3650 阅读 · 2 评论 -
pytorch和torchvision的安装
在pycharm中使用清华或中科院的源,找不到匹配的版本,因此使用下面的一行命令即可安装。在命令行中执行:pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html...原创 2020-02-29 16:48:02 · 3158 阅读 · 0 评论