Trait dfdx::nn::ResetParams

pub trait ResetParams<E: Dtype, D: Device<E>>: TensorCollection<E, D> {
    // Provided methods
    fn reset_params(&mut self) { ... }
    fn try_reset_params(&mut self) -> Result<(), D::Err> { ... }
}

Reset a module’s parameters with their default reset function:

type Model = Linear<2, 5>;
let mut model = dev.build_module::<Model, f32>();
model.reset_params();

Provided Methods

fn reset_params(&mut self)

Reset all a model’s parameters.

fn try_reset_params(&mut self) -> Result<(), D::Err>

Reset all a model’s parameters.

Implementors

impl<E: Dtype, D: Device<E>, M: TensorCollection<E, D>> ResetParams<E, D> for M