Pytorch
文章平均质量分 80
一颗磐石
这个作者很懒,什么都没留下…
展开
-
nn.Sequential()
nn.Sequential()nn.Sequential()介绍一个序列容器,用于搭建神经网络的模块被按照被传入构造器的顺序添加到nn.Sequential()容器中。除此之外,一个包含神经网络模块的OrderedDict也可以被传入nn.Sequential()容器中。利用nn.Sequential()搭建好模型架构,模型前向传播时调用forward()方法,模型接收的输入首先被传入nn.Sequential()包含的第一个网络模块中。然后,第一个网络模块的输出传入第二个网络模块作为输入,按照顺序依次原创 2022-04-15 16:14:15 · 66356 阅读 · 4 评论 -
深度学习之动态调整学习率LR
深度学习之动态调整学习率LR什么是学习率为什么要动态调整学习率基于Pytorch动态调整学习率的几种常见方法原创 2022-03-26 14:14:48 · 8082 阅读 · 0 评论 -
torch.argmax(input, dim, keepdim=False)
文章目录导读官方文档地址参数解析举例演示导读最近有时间看一些目标检测项目的代码(基于Pytorch),里边很多Pytorch的相关操作都忘记了,特来此记录一下,用以加深记忆,而且还能以备一样处境的同学前来查询。今天的主角是torch.argmax(input, dim, keepdim=False)。官方文档地址https://pytorch.org/docs/stable/generated/torch.argmax.htmltorch.argmax(input) → LongTensor原创 2022-03-08 18:17:12 · 2634 阅读 · 0 评论 -
Pytorch中的contiguous理解
最近在使用Pytorch搭建网络的时候,遇到一个错误,特来此记录一下,给路过的朋友设置踩坑预警。报错信息RuntimeError: input is not contiguous按照常规操作,谷歌翻译搞起,得到结论:RuntimeError:输入不连续看到翻译结果,我…一脸懵,输入不连续是什么鬼。为了省时间,还是选择冲浪Debug。报错原因先补充一些基础知识:我们知道一个张量tensor有shape和stride这些属性,那么这些属性之间的区别和联系是怎样的?以二维矩阵为例,shape =原创 2022-01-13 19:53:15 · 683 阅读 · 0 评论