Function tch::nn::rms_prop

source · 
pub fn rms_prop(
    alpha: f64,
    eps: f64,
    wd: f64,
    momentum: f64,
    centered: bool
) -> RmsProp
Expand description

Creates the configuration for the RmsProp optimizer.