#
我就廢話不多說了，直接上代碼吧！ class Net(nn.Module): def __init__(self , model): super(Net, self).__init__
前言 最近使用PyTorch感覺妙不可言，有種當初使用Keras的快感，而且速度還不慢。各種設計直接簡潔，方便研究，比tensorflow的臃腫好多了。今天讓我們來談談PyTorch的預訓練，主要是
一個繼承nn.module的model它包含一個叫做children()的函數，這個函數可以用來提取出model每一層的網絡結構，在此基礎上進行修改即可，修改方法如下(去除后兩層)： resnet
已經有了一個預訓練的模型，我需要從其中取出某一層，把該層的weights和biases賦值到新的網絡結構中，可以使用tensorflow中的pywrap_tensorflow(用來讀取預訓練模型的參數
