Pytorch 實現(xiàn)凍結(jié)指定卷積層的參數(shù)
更新時間：2020年01月06日 09:36:19 作者：陳軍2號
今天小編就為大家分享一篇Pytorch 實現(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 實現(xiàn)凍結(jié)指定卷積層的參數(shù)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python機器學(xué)習(xí)中實現(xiàn)距離和相似性計算詳解
這篇文章主要為大家詳細(xì)介紹了Python機器學(xué)習(xí)中實現(xiàn)距離和相似性計算的方法，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2023-03-03
這篇文章主要介紹了Python中關(guān)于set的基本用法，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-04-04
Python預(yù)測2020高考分?jǐn)?shù)和錄取情況
這篇文章主要介紹了Python預(yù)測2020高考分?jǐn)?shù)和錄取情況可能是這樣,本文通過示例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-07-07
python 如何將帶小數(shù)的浮點型字符串轉(zhuǎn)換為整數(shù)
在python中如何實現(xiàn)將帶小數(shù)的浮點型字符串轉(zhuǎn)換為整數(shù)呢？今天小編就為大家介紹一下解決方案，希望對大家有所幫助。一起跟隨小編過來看看吧2021-05-05
基于python實現(xiàn)cdn日志文件導(dǎo)入mysql進行分析
這篇文章主要介紹了基于python實現(xiàn)cdn日志文件導(dǎo)入mysql進行分析，本文以阿里云CDN日志作為輔助查詢數(shù)據(jù)展開主題內(nèi)容，其它云平臺大同小異，需要的小伙伴可以參考一下2022-05-05
這篇文章主要為大家詳細(xì)介紹了Flask Web開發(fā)入門之文件上傳的相關(guān)資料，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-08-08
這篇文章主要介紹了基于Numba提高python運行效率過程解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-03-03
心情不好的時候，來用Python和Turtle庫畫棵櫻花樹送給自己吧，自己也要好好愛自己才對！文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以動手試一試2022-02-02
這篇文章主要為大家詳細(xì)介紹了python使用多線程編寫tcp客戶端程序，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-09-09
range()是python內(nèi)置函數(shù)它能返回一系列連續(xù)增加的整數(shù)，它的工作方式類似于分片，可以生成一個列表對象,這篇文章主要介紹了Python?range()?函數(shù)用法,需要的朋友可以參考下2023-03-03