Menambahkan lapisan konvolusional baru

Pimpinan proyek Anda memberikan sebuah model CNN baru. Mari kita lihat arsitektur model tersebut dan tambahkan satu lapisan konvolusional baru.

Model tersedia sebagai CNNModel. Paket torch dan torch.nn sebagai nn sudah diimpor.

Latihan ini adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk latihan

Instansiasikan model dari kelas CNNModel dan akses lapisan-lapisan konvolusionalnya.
Buat lapisan konvolusional baru dengan in_channels sama dengan out_channels milik lapisan yang ada, out_channels diatur ke 32, serta stride dan padding keduanya 1, dan kernel_size bernilai 3; tetapkan ke conv2.
Tambahkan (append) lapisan baru ke model dengan nama "conv2".

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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)

Edit dan Jalankan Kode