PyTorch中定義模型時，有時候會遇到self.register_buffer('name', Tensor)的操作，該方法的作用是定義一組參...
投稿
收錄了8篇文章 · 3人關注
-
PyTorch nn.Module中的self.register_buffer()解析
-
PyTorch獲取自帶模型的中間特征圖
本文首發(fā)自【簡書】用戶【西北小生_】的博客，轉載請注明出處！ PyTorch之HOOK——獲取神經網絡特征和梯度的有效工具[https://ww...
-
torchsummary：計算神經網絡模型各層輸出特征圖尺寸及參數量
之前寫過一篇自動計算模型參數量、FLOPs、乘加數以及所需內存等數據[http://www.itdecent.cn/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模型保存與加載，并在加載的模型基礎上繼續(xù)訓練[https://www.jianshu.c...
-
Pytorch模型保存與加載，并在加載的模型基礎上繼續(xù)訓練
pytorch保存模型非常簡單，主要有兩種方法： 只保存參數；（官方推薦） 保存整個模型 (結構+參數)。由于保存整個模型將耗費大量的存儲，故官...
-
Pytorch中的學習率衰減及其用法學習率衰減是一個非常有效的煉丹技巧之一，在神經網絡的訓練過程中，當accuracy出現震蕩或loss不再下降時，進行適當的學習率衰減是一個行之有...
-
torch.squeeze()和torch.unsqueeze()
1. torch.squeeze(tensor) 和numpy等庫函數中的squeeze()函數作用一樣，torch.squeeze()函數的作...
{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》（ 分享自 @簡書 ）','http://www.itdecent.cn/c/541f8305622c?utm_campaign=maleskine&utm_content=collection&utm_medium=reader_share&utm_source=weibo','頁面編碼gb2312|utf-8默認gb2312'));){kind=link}