Model dua masukan
Dengan data yang sudah siap, saatnya membangun arsitektur model dengan dua masukan! Untuk melakukannya, Anda akan menyiapkan kelas model dengan metode berikut:
.__init__(), tempat Anda mendefinisikan sub-jaringan dengan mengelompokkan layer; di sinilah Anda mendefinisikan dua layer untuk memproses dua masukan, serta classifier yang mengembalikan skor klasifikasi untuk setiap kelas.forward(), tempat Anda meneruskan kedua masukan melalui sub-jaringan yang telah didefinisikan, menggabungkan keluarannya, lalu meneruskannya ke classifier.
torch.nn sudah diimpor sebagai nn untuk Anda. Mari kita mulai!
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__()
# Define sub-networks as sequential models
____ = ____(
nn.Conv2d(1, 16, kernel_size=3, padding=1),
nn.MaxPool2d(kernel_size=2),
nn.ELU(),
nn.Flatten(),
nn.Linear(16*32*32, 128)
)
____ = ____(
nn.Linear(30, 8),
nn.ELU(),
)
____ = ____(
nn.Linear(128 + 8, 964),
)