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

Exercițiu

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

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

Instrucțiuni

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