#
當使用pytorch寫網絡結構的時候,本人發現在卷積層與第一個全連接層的全連接層的input_features不知道該寫多少?一開始本人的做法是對著pytorch官網的公式推,但是總是算錯。 后來發現
1.簡介(torch.nn下的) 卷積層主要使用的有3類,用于處理不同維度的數據 參數 Parameters: in_channels(int) – 輸入信號的通道 out_channels(int)
由于研究關系需要自己手動給卷積層初始化權值,但是好像博客上提到的相關文章比較少(大部分都只提到使用nn.init里的按照一定分布初始化方法),自己參考了下Pytorch的官方文檔,發現有兩種方法吧。
python代碼 for i, para in enumerate(self._net.module.features.parameters()): if i < 16: