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
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
____ = ____
____ = ____