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 merupakan bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Instruksi 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 langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# 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