pytorch【模型模塊】一、模型創(chuàng)建步驟與nn.Module

一、前言

機(jī)器學(xué)習(xí)模型主要有以下幾個(gè)大的模塊：

前面的內(nèi)容是關(guān)于數(shù)據(jù)模塊，這里的內(nèi)容是關(guān)于模型模塊。

二、模型創(chuàng)建步驟

LeNet由各層組成，一個(gè)LeNet是一個(gè)nn.Module，一個(gè)卷積層是一個(gè)nn.Module，所以一個(gè)nn.Module是可以包含很多子nn.Module的。

（這里對(duì)應(yīng)代碼lesson-10）

三、nn.Module

所有的模型模塊都是繼承自nn.Module，所以需詳細(xì)了解nn.Module這個(gè)類

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容