[深度學習]Pytorch中RNN模型小結

Pytorch中所有模型分爲構造參數和輸入和輸出構造參數兩種類型。  模型構造參數主要限定了網絡的結構,如對循環網絡,則包括輸入維度、隱層\輸出維度、層數;對卷積網絡,無論卷積層還是池化層,都不關心輸入維度,其構造方法只涉及卷積核大小\步長等。這裏的參數決定了模型持久化後的大小.  輸入和輸出的構造參數一般和模型訓練相關,都需指定batch大小,seq大小(循環網絡)\chanel大小(卷積網絡)
相關文章
相關標籤/搜索