ResNet解析(pytorch源碼)

首先放一張各層的圖片，整體分爲4個layer, pytorch中也是這麼分的 然後這是兩種設計方式，左邊的是用於18,34層的，這樣參數多，右面這種設計方式參數少，適用於更深度的 這裏是這兩個基本塊的代碼，然後ResNet中把這些塊連接起來就可以組成網絡。最重要的代碼就是下面這3部分，然後通過傳入不同的參數給ResNet就可以建立各種經典層的網絡了。 class BasicBlock(nn.Mod

>>阅读原文<<