快捷導(dǎo)航

Pytorch 實(shí)現(xiàn)凍結(jié)指定卷積層的參數(shù)

更新時(shí)間：2020年01月06日 09:36:19 作者：陳軍2號(hào)

今天小編就為大家分享一篇Pytorch 實(shí)現(xiàn)凍結(jié)指定卷積層的參數(shù)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

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ù)就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

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