1. Learn
    2. /
  2. Курси
    3. /
  3. Середній рівень Deep Learning з PyTorch
Connected

вправа

Навчання моделей із кількома виходами

Під час навчання моделей із кількома виходами важливо правильно визначити функцію втрат.

У цьому прикладі модель повертає два виходи: передбачення для абетки та для символу. Для кожного з них є відповідні еталонні мітки, завдяки яким ви зможете обчислити дві окремі втрати: одну — за неправильні класифікації абетки, іншу — за неправильну класифікацію символу. Оскільки в обох випадках ідеться про задачу багатокласової класифікації, щоразу можна застосувати втрату перехресної ентропії (Cross-Entropy).

Однак градієнтний спуск може оптимізувати лише одну функцію втрат. Тому ви визначите сукупну втрату як суму втрат для абетки й для символу.

Інструкції

100 XP
  • Обчисліть втрату для класифікації абетки та присвойте її змінній loss_alpha.
  • Обчисліть втрату для класифікації символу та присвойте її змінній loss_char.
  • Обчисліть загальну втрату як суму двох часткових втрат і присвойте її змінній loss.