深度学习
cactus233
这个作者很懒,什么都没留下…
展开
-
统计深度学习模型的参数和显存占用(pytorch)
这里介绍两种方法1、使用相关的库torchsummaryfrom torchsummary import summarynet=net.to(torch.device("cpu"))#or cudasummary(net,(4,228,912),device="cpu") #or cuda统计结果比较详细,参数量、浮点数计算量、中间变量、train的变量数、保持不变的变量数,每...原创 2019-12-23 16:14:38 · 5151 阅读 · 0 评论 -
卷积神经网络的dropout layer
1、dropout是什么深度学习领域大神Hinton,在2012年文献:《Improving neural networks by preventingco-adaptation of feature detectors》提出了,在每次训练的时候,让一半的特征检测器停过工作,这样可以提高网络的泛化能力,Hinton又把它称之为dropout。 Dropout用网络训练时工作变化来定义...原创 2019-05-29 01:21:20 · 7957 阅读 · 0 评论