Эксперименты с моментом
В этом упражнении ваша цель — найти оптимальное значение момента, при котором оптимизатор сможет найти минимум следующей невыпуклой функции \(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)