PyTorch中定义模型时，有时候会遇到self.register_buffer('name', Tensor)的操作，该方法的作用是定义一组参...
投稿
收录了8篇文章 · 3人关注
-
PyTorch nn.Module中的self.register_buffer()解析
-
PyTorch获取自带模型的中间特征图
本文首发自【简书】用户【西北小生_】的博客，转载请注明出处！ PyTorch之HOOK——获取神经网络特征和梯度的有效工具[https://ww...
-
torchsummary：计算神经网络模型各层输出特征图尺寸及参数量
之前写过一篇自动计算模型参数量、FLOPs、乘加数以及所需内存等数据[https://www.jianshu.com/p/ca6da98b2ec...
-
PyTorch中的model.modules(), model.children(), model.named_children(), model.parameters(), model.nam...
本文通过一个例子实验来观察并讲解PyTorch中model.modules(), model.named_modules(), model.ch...
-
PyTorch模型保存深入理解
前面写过一篇PyTorch保存模型的文章：Pytorch模型保存与加载，并在加载的模型基础上继续训练[https://www.jianshu.c...
-
Pytorch模型保存与加载，并在加载的模型基础上继续训练
pytorch保存模型非常简单，主要有两种方法： 只保存参数；（官方推荐） 保存整个模型 (结构+参数)。由于保存整个模型将耗费大量的存储，故官...
-
Pytorch中的学习率衰减及其用法学习率衰减是一个非常有效的炼丹技巧之一，在神经网络的训练过程中，当accuracy出现震荡或loss不再下降时，进行适当的学习率衰减是一个行之有...
-
torch.squeeze()和torch.unsqueeze()
1. torch.squeeze(tensor) 和numpy等库函数中的squeeze()函数作用一样，torch.squeeze()函数的作...
专题公告
Pytorch学习笔记
{var%20f='http://v.t.sina.com.cn/share/share.php?appkey=1881139527',u=z||d.location,p=['&url=',e(u),'&title=',e(t||d.title),'&source=',e(r),'&sourceUrl=',e(l),'&content=',c||'gb2312','&pic=',e(p||'')].join('');function%20a(){if(!window.open([f,p].join(''),'mb',['toolbar=0,status=0,resizable=1,width=440,height=430,left=',(s.width-440)/2,',top=',(s.height-430)/2].join('')))u.href=[f,p].join('');};if(/Firefox/.test(navigator.userAgent))setTimeout(a,0);else%20a();})(screen,document,encodeURIComponent,'','','https://upload.jianshu.io/collections/images/1802572/pytorch.png', '推荐专题《Pytorch》（ 分享自 @简书 ）','https://wsa.jianshu.io/c/541f8305622c?utm_campaign=maleskine&utm_content=collection&utm_medium=reader_share&utm_source=weibo','页面编码gb2312|utf-8默认gb2312'));){kind=link}