尝试不同的动量参数

在本练习中，您的目标是在 20 步内找到能使优化器收敛到以下非凸函数 \(x^{4} + x^{3} - 5x^{2}\) 最小值的最佳动量（momentum）。您将尝试两个不同的动量取值。本题的学习率固定为 0.01。

已为您提供可调用的 optimize_and_plot() 函数，它接收 momentum 参数作为输入。该函数会运行 SGD 优化器的 20 步并显示结果。