#

全連接層

  • pytorch神經(jīng)網(wǎng)絡(luò)之卷積層與全連接層參數(shù)的設(shè)置方法

    當(dāng)使用pytorch寫(xiě)網(wǎng)絡(luò)結(jié)構(gòu)的時(shí)候，本人發(fā)現(xiàn)在卷積層與第一個(gè)全連接層的全連接層的input_features不知道該寫(xiě)多少？一開(kāi)始本人的做法是對(duì)著pytorch官網(wǎng)的公式推，但是總是算錯(cuò)。 后來(lái)發(fā)現(xiàn)

    作者：嘿芝麻
    2020-10-21 23:33:02