ResNet解析(pytorch源碼)

首先放一張各層的圖片,總體分爲4個layer, pytorch中也是這麼分的python 而後這是兩種設計方式,左邊的是用於18,34層的,這樣參數多,右面這種設計方式參數少,適用於更深度的網絡 這裏是這兩個基本塊的代碼,而後ResNet中把這些塊鏈接起來就能夠組成網絡。最重要的代碼就是下面這3部分,而後經過傳入不一樣的參數給ResNet就能夠創建各類經典層的網絡了。app class Basic
相關文章
相關標籤/搜索