MulaiMulai sekarang secara gratis

Arsitektur model dua keluaran

Dalam latihan ini, Anda akan membangun arsitektur jaringan saraf dengan banyak keluaran yang mampu memprediksi karakter dan alfabet.

Ingat struktur umumnya: dalam metode .__init__(), Anda mendefinisikan layer yang akan digunakan pada proses forward nanti. Dalam metode forward(), Anda terlebih dahulu meneruskan citra masukan melalui beberapa layer untuk memperoleh embedding-nya, yang selanjutnya diberikan ke dua layer pengklasifikasi terpisah, masing-masing untuk setiap keluaran.

torch.nn sudah diimpor dengan alias biasanya, jadi mari kita bangun modelnya!

Latihan ini adalah bagian dari kursus

Deep Learning Lanjutan dengan PyTorch

Lihat Kursus

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

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
        ____ = ____
        ____ = ____
Edit dan Jalankan Kode