1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Середній рівень Deep Learning з PyTorch
Connected

Bài tập

Цикл навчання класифікатора зображень

Час тренувати класифікатор зображень! Ви використаєте визначений раніше Net і навчите його розрізняти сім типів хмар.

Щоб визначити функцію втрат і оптимізатор, скористайтеся функціями з torch.nn та torch.optim, які імпортовано для вас як nn і optim відповідно. Вам не потрібно змінювати сам цикл навчання: він такий самий, як ті, що ви вже писали, лише з додатковою логікою для виведення значення втрат під час навчання.

Hướng dẫn

100 XP
  • Створіть модель, використавши ваш клас Net із num_classes, встановленим у 7, і присвойте її змінній net.
  • Визначте функцію втрат як перехресну ентропію й присвойте її criterion.
  • Визначте оптимізатор Adam, передавши йому параметри моделі та швидкість навчання 0.001, і присвойте його optimizer.
  • Розпочніть навчальний for-цикл, ітеруючись за тренувальними images та labels з dataloader_train.