新增一個卷積層
你的專案負責人提供了一個新的 CNN 模型。先來看看這個模型的架構,並在其中加上一個新的卷積層。
模型已以 CNNModel 名稱提供。torch 與 torch.nn(縮寫為 nn)已經匯入。
本練習屬於課程
使用 PyTorch 進行影像深度學習
練習說明
- 由
CNNModel類別實例化一個模型,並存取其中的卷積層。 - 建立一個新的卷積層:其
in_channels等於現有該層的out_channels,out_channels設為 32,stride與padding都設為 1,且kernel_size為 3;將它指定給conv2。 - 把這個新層加到模型中,名稱設為
"conv2"。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Create a model
model = ____
print("Original model: ", model)
# Create a new convolutional layer
conv2 = ____
# Append the new layer to the model
model.____(____)
print("Extended model: ", model)