mirror of
https://github.com/huggingface/candle.git
synced 2025-06-14 09:57:10 +00:00
Expose AdamW parameters (#1449)
* Expose AdamW parameters * Use reference
This commit is contained in:
@ -190,4 +190,12 @@ impl AdamW {
|
||||
};
|
||||
Self::new(vars, params)
|
||||
}
|
||||
|
||||
pub fn params(&self) -> &ParamsAdamW {
|
||||
&self.params
|
||||
}
|
||||
|
||||
pub fn set_params(&mut self, params: ParamsAdamW) {
|
||||
self.params = params;
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user