1. Обучение
    2. /
  2. Курса
    3. /
  3. Введение в глубокое обучение с PyTorch
Connected

упражнение

Эксперименты с dropout

Dropout помогает бороться с переобучением, случайным образом обнуляя часть выходных значений во время обучения. В этом упражнении вы построите простую нейронную сеть с dropout и посмотрите, как она ведёт себя в режимах обучения и оценки.

Пакет torch.nn заранее импортирован как nn, переменная features уже определена.

Инструкции 1/2

undefined XP
    1
    2
  • Добавьте в модель слой dropout с вероятностью обнуления 50%.
  • Переведите модель в режим обучения и получите предсказания на основе features, сохранив результат в переменную output_train.