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.