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使用Redis實現(xiàn)作業(yè)調(diào)度系統(tǒng)(超簡單)
Redis作為內(nèi)存數(shù)據(jù)庫的一個典型代表，已經(jīng)在很多應(yīng)用場景中被使用，這里僅就Redis的pub/sub功能來說說怎樣通過此功能來實現(xiàn)一個簡單的作業(yè)調(diào)度系統(tǒng)。這里只是想展現(xiàn)一個簡單的想法，所以還是有很多需要考慮的東西沒有包括在這個例子中，比如錯誤處理，持久化等2016-03-03
這篇文章主要給大家介紹了Python基于面向?qū)ο笞鲆粋€文件夾整理工具,文中的示例代碼講解詳細,具有一定的參考價值,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-02-02
freeswitch支持多種語言的業(yè)務(wù)開發(fā)，包括C/C++，java，python，js，lua，Golang等等。freeswitch在使用python做業(yè)務(wù)開發(fā)時，有倆種接入方式，一種是ESL接口，另一種是mod_python模塊。本文主要介紹的是fs內(nèi)部的mod_python語言支持模塊，需要的朋友可以參考下面文章內(nèi)容2021-09-09
這篇文章主要為大家詳細介紹了如何利用Python編寫一個自動化部署工具，文中的示例代碼講解詳細，具有一定的學(xué)習(xí)價值，需要的可以了解一下2023-07-07
python使用docx模塊讀寫docx文件的方法與docx模塊常用方法詳解
這篇文章主要介紹了python使用docx模塊讀寫docx文件的方法與docx模塊常用方法詳解,需要的朋友可以參考下2020-02-02
這篇文章主要介紹了Python底層封裝實現(xiàn)方法詳解,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-01-01
什么！你不會背圓周率（鄙夷的眼神） 3.1415926535 8979323846 26433... 但是，我會算啊，本文用一個簡單的python代碼，教你計算圓周率2021-08-08
本文主要介紹了如何使用OpenCV和EAST文本檢測器檢測圖像中的文本，以便大家可以在自己的應(yīng)用程序中應(yīng)用文本檢測。感興趣的同學(xué)可以關(guān)注一下2021-11-11
Python asyncore socket客戶端實現(xiàn)方法詳解
這篇文章主要介紹了Python asyncore socket客戶端實現(xiàn)方法，asyncore庫是python的一個標準庫，提供了以異步的方式寫入套接字服務(wù)的客戶端和服務(wù)器的基礎(chǔ)結(jié)構(gòu)2022-12-12
這篇文章主要介紹了django上傳圖片并生成縮略圖方法示例,具有一定借鑒價值，需要的朋友可以參考下。2017-12-12