評價此頁

torch.optim.rmsprop.rmsprop#

torch.optim.rmsprop.rmsprop(params, grads, square_avgs, grad_avgs, momentum_buffer_list, state_steps, foreach=None, maximize=False, differentiable=False, capturable=False, has_complex=False, *, lr, alpha, eps, weight_decay, momentum, centered)[原始碼]#

執行 RMSprop 演算法計算的函式式 API。

詳情請參閱 RMSProp