Блок класифікатора

Ваше наступне завдання — створити блок класифікатора, який замінить початковий класифікатор VGG16. Ви вирішили використати блок із двома повнозв'язними шарами з активацією ReLU між ними.

vgg_model та input_dim, які ви визначили у попередній вправі, доступні у вашому робочому середовищі, а torch і torchvision.models вже імпортовано.

Ця вправа є частиною курсу

Глибоке навчання для зображень із PyTorch

Переглянути курс

Інструкції до вправи

Створіть змінну num_classes з кількістю класів, припускаючи, що ви розв'язуєте задачу виявлення лише котів і собак.
Створіть послідовний блок за допомогою nn.Sequential.
Створіть лінійний шар із in_features, встановленим у input_dim.
Додайте вихідні ознаки до останнього шару класифікатора.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

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

Редагувати та запускати код