Blok classifier

Tugas Anda berikutnya adalah membuat blok classifier yang akan menggantikan classifier VGG16 asli. Anda memutuskan untuk menggunakan blok dengan dua lapisan fully connected dengan aktivasi ReLU di antaranya.

vgg_model dan input_dim yang Anda definisikan pada latihan sebelumnya tersedia di ruang kerja Anda, dan torch serta torchvision.models sudah diimpor.

Latihan ini adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk latihan

Buat variabel num_classes dengan jumlah kelas dengan asumsi Anda hanya mendeteksi kucing dan anjing.
Buat blok sequential menggunakan nn.Sequential.
Buat lapisan linear dengan in_features disetel ke input_dim.
Tambahkan fitur keluaran ke lapisan terakhir classifier.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Create a variable with the number of classes
____
    
# Create a sequential block
classifier = ____(
	# Create a linear layer with input features
	____(____, 512),
	nn.ReLU(),
	# Add the output dimension to the classifier
	nn.Linear(512, ____),
)

Edit dan Jalankan Kode