1. Learn
    2. /
  2. Курси
    3. /
  3. Глибоке навчання для зображень із PyTorch
Connected

Вправа

Завантаження попередньо натренованої моделі

Ви створюєте застосунок для розмічення зображень із соціальних мереж. Це завдання потребує високої точності та швидкості. Ви використаєте попередньо натреновану модель ResNet18, щоб визначати класи зображень.

Інструкції

100 XP
  • Імпортуйте resnet18 і ResNet18_Weights з torchvision.models.
  • Створіть екземпляр model за допомогою resnet18(), передавши параметр weights як weights.
  • Переведіть model у режим оцінювання.
  • Ініціалізуйте вхідні перетворення та присвойте їх змінній transform.