Function tch::nn::adam[][src]

pub fn adam(beta1: f64, beta2: f64, wd: f64) -> Adam

Creates the configuration for the Adam optimizer.