НачатьНачать бесплатно

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

В этом упражнении ваша цель — найти оптимальное значение момента, при котором оптимизатор сможет найти минимум следующей невыпуклой функции \(x^{4} + x^{3} - 5x^{2}\) за 20 шагов. Вы попробуете два разных значения момента. Скорость обучения для этой задачи фиксирована на уровне 0,01.

Вам предоставлена функция optimize_and_plot(), которая принимает на вход параметр momentum. Эта функция выполнит 20 шагов оптимизатора SGD и отобразит результаты.

Это упражнение является частью курса

Введение в глубокое обучение с PyTorch

Посмотреть курс

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Try a first value for momentum
mom0 = ____
optimize_and_plot(momentum=mom0)
Редактировать и запускать код