pytorch基礎教程-模型編寫（二）

對於一些重複的layer可以採用nn.Sequential先進行封裝layer，然後調用_make_layer_進行實現 以ResNet18/34爲例 模型圖： ResNet中將一個跨層直連的單元叫做Residual block,其結構如下，當輸入和輸出的通道不一致的時候，需要一個專門的單元將二者轉成一致，使其可以相加。 實現： class BasicBlock(nn.Module): e

>>阅读原文<<