LinDragon 筆記圖書館2026年10月
Journal of CNN

卷積神經網路 · 優化器 · 學習率調度

Research Note

RMSprop

RMSprop

可以先把它當成「自動調整 learning rate 的加強版 SGD」

RMSprop 會根據「最近一段時間的梯度大小」,自動幫每個參數調整步伐大小

特點:Adam的前身,適合RNN

直觀感覺:

所以它特別適合:

optimizer = optim.RMSprop(
    model.parameters(),
    lr=1e-3,           # 學習率
    alpha=0.99,        # 平滑常數
    eps=1e-8,          # 數值穩定性
    weight_decay=0,    # 權重衰減
    momentum=0         # 動量
)

和 Adam / SGD 比較

優點:

缺點:

適用場景:


相關筆記