Struct ParamGroupBuilder 

pub struct ParamGroupBuilder { /* private fields */ }

Builder for creating parameter groups with various options

Implementations

impl ParamGroupBuilder

pub fn new(lr: f32) -> Self

Create a new parameter group builder

pub fn params(self, params: Vec<Arc<RwLock<Tensor>>>) -> Self

Add parameters to the group

pub fn add_param(self, param: Arc<RwLock<Tensor>>) -> Self

Add a single parameter to the group

pub fn weight_decay(self, weight_decay: f32) -> Self

Set weight decay

pub fn eps(self, eps: f32) -> Self

Set epsilon

pub fn option(self, key: String, value: f32) -> Self

Set a custom option

pub fn from_options(self, options: &OptimizerOptions) -> Self

Set options from OptimizerOptions

pub fn build(self) -> ParamGroup

Build the parameter group

Trait Implementations

impl Debug for ParamGroupBuilder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.