開始使用免費開始

雙輸出模型架構

在這個練習中,你將建立一個多輸出的類神經網路架構,用來同時預測字元與字母表。

回顧一般結構:在 .__init__() 方法中,你會先定義稍後前向傳遞要用到的層。接著在 forward() 方法中,先把輸入影像傳過幾個層以取得其嵌入向量(embedding),再把這個嵌入向量分別送進兩個獨立的分類器層,各對應一個輸出。

torch.nn 已用慣用別名匯入,現在就來建立模型吧!

本練習屬於課程

Intermediate Deep Learning with PyTorch

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        self.image_layer = nn.Sequential(
            nn.Conv2d(1, 16, kernel_size=3, padding=1),
            nn.MaxPool2d(kernel_size=2),
            nn.ELU(),
            nn.Flatten(),
            nn.Linear(16*32*32, 128)
        )
        # Define the two classifier layers
        ____ = ____
        ____ = ____
編輯並執行程式碼