#

nn.modulelist

  • 對(duì)Pytorch中nn.ModuleList 和 nn.Sequential詳解

    簡(jiǎn)而言之就是,nn.Sequential類似于Keras中的貫序模型,它是Module的子類,在構(gòu)建數(shù)個(gè)網(wǎng)絡(luò)層之后會(huì)自動(dòng)調(diào)用forward()方法,從而有網(wǎng)絡(luò)模型生成。而nn.ModuleList僅僅

    作者:ustc_lijia
    2020-09-04 05:47:05