開始使用免費開始

新增一個卷積層

你的專案負責人提供了一個新的 CNN 模型。先來看看這個模型的架構,並在其中加上一個新的卷積層。

模型已以 CNNModel 名稱提供。torchtorch.nn(縮寫為 nn)已經匯入。

本練習屬於課程

使用 PyTorch 進行影像深度學習

檢視課程

練習說明

  • CNNModel 類別實例化一個模型,並存取其中的卷積層。
  • 建立一個新的卷積層:其 in_channels 等於現有該層的 out_channelsout_channels 設為 32,stridepadding 都設為 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)
編輯並執行程式碼