1. Nauka
    2. /
  2. Kursy
    3. /
  3. 使用 PyTorch 的深度学习入门
Connected

ćwiczenie

尝试不同的动量参数

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

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

Instrukcje 1/2

undefined XP
  • 1
    • 先选择一个动量值，使优化器卡在第一个局部最小值。
  • 2
    • 再选择第二个动量值，使优化器找到全局最优解。