Pytorch 實(shí)現(xiàn)凍結(jié)指定卷積層的參數(shù)
python代碼
for i, para in enumerate(self._net.module.features.parameters()): if i < 16: para.requires_grad = False else: para.requires_grad = True # Solver. # self._solver = torch.optim.SGD( # self._net.parameters(), lr=self._options['base_lr'], # momentum=0.9, weight_decay=self._options['weight_decay']) self._solver = torch.optim.SGD( self._net.module.parameters(), lr=self._options['base_lr'], momentum=0.9, weight_decay=self._options['weight_decay'])
分析
通過for循環(huán)將需要凍結(jié)的layer的requires_grad屬性設(shè)置為False
以上這篇Pytorch 實(shí)現(xiàn)凍結(jié)指定卷積層的參數(shù)就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python使用Redis實(shí)現(xiàn)作業(yè)調(diào)度系統(tǒng)(超簡單)
Redis作為內(nèi)存數(shù)據(jù)庫的一個(gè)典型代表,已經(jīng)在很多應(yīng)用場景中被使用,這里僅就Redis的pub/sub功能來說說怎樣通過此功能來實(shí)現(xiàn)一個(gè)簡單的作業(yè)調(diào)度系統(tǒng)。這里只是想展現(xiàn)一個(gè)簡單的想法,所以還是有很多需要考慮的東西沒有包括在這個(gè)例子中,比如錯(cuò)誤處理,持久化等2016-03-03Python基于面向?qū)ο笞鲆粋€(gè)文件夾整理工具
這篇文章主要給大家介紹了Python基于面向?qū)ο笞鲆粋€(gè)文件夾整理工具,文中的示例代碼講解詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-02-02python使用docx模塊讀寫docx文件的方法與docx模塊常用方法詳解
這篇文章主要介紹了python使用docx模塊讀寫docx文件的方法與docx模塊常用方法詳解,需要的朋友可以參考下2020-02-02實(shí)現(xiàn)用python算法計(jì)算圓周率的小訣竅
什么!你不會(huì)背圓周率(鄙夷的眼神) 3.1415926535 8979323846 26433... 但是,我會(huì)算啊,本文用一個(gè)簡單的python代碼,教你計(jì)算圓周率2021-08-08Python+Opencv文本檢測(cè)的實(shí)現(xiàn)
本文主要介紹了如何使用OpenCV和EAST文本檢測(cè)器檢測(cè)圖像中的文本,以便大家可以在自己的應(yīng)用程序中應(yīng)用文本檢測(cè)。感興趣的同學(xué)可以關(guān)注一下2021-11-11Python asyncore socket客戶端實(shí)現(xiàn)方法詳解
這篇文章主要介紹了Python asyncore socket客戶端實(shí)現(xiàn)方法,asyncore庫是python的一個(gè)標(biāo)準(zhǔn)庫,提供了以異步的方式寫入套接字服務(wù)的客戶端和服務(wù)器的基礎(chǔ)結(jié)構(gòu)2022-12-12