Pytorch 各種奇葩古怪的使用方法

2021-09-19 21:08:09 字數 1139 閱讀 4677

不間斷更新。。。

增加layer很方便,可以使用model.add_module('layer name', layer)

刪減layer很少用的到,之所以我會有這麼乙個需求,是因為我需要使用vgg做遷移學習,而且需要修改最後的輸出。

而vgg由兩個部分組成:featuresclassifier,這兩個部分都是torch.nn.sequential,所以不能單獨對其中某一層做修改。

而如果對整個sequential做修改,那麼這個模型的引數會被初始化,而我又需要保留這些引數,所以才想到是否有辦法把最後一層fc刪掉,重新再填加乙個不就行了?具體方法如下:

以vgg16為例,假設我們現在只需要對classifier的最後一層全連線層的輸出做修改:

model = models.vgg16(pretrained=true)
先看一下未做修改之前的classifier的引數:

model.add_module('fc', torch.nn.linear(4096, out_num)) # out_num是你希望輸出的數量
此時我們看一下model以及classifier的引數有什麼變化:

marsggbo原創

2018-4-10

各種奇葩的貸款產品

關注 番茄風控大資料 獲取更多資料分析與風控大資料的實用乾貨。移動網際網路的發展給金融產品提供了乙個絕好的變現場景。今天我們梳理了些大眾化的貸款產品,也有一些非正常類的產品,看完之後,只能佩服腦洞大開的產品經理,這種市場嗅覺沒得說。房貸車貸 正常指數 中國家庭的兩座大山,房貸和車貸。當然還有孩貸。傳...

pytorch的各種錯誤集錦

no module named swigfaiss 原因 沒有安裝依賴包,而且我直接用pip install faiss安裝,沒採用官方推薦的conda方法 解決方法 1 刪除原本安裝的faiss 2.conda install mkl,openblas swig 3.用conda安裝faiss c...

搭建集群時遇到各種奇葩問題的大招

用自己電腦搭建hadoop集群的時候總是會遇到各種奇葩的問題,因為虛擬機器總是會比真實的環境差很多的,有些奇葩的問題真是讓我鬱悶不行不行的 像這個問題,搭了三遍的集群遇見各種奇葩問題,最後找了乙個終極大招 但是,但是,前提是,新搭的集群,如果hdfs裡面已經有資料了,那就別用了,因為會把你的hado...