huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose AdamW parameters (#1449)

Browse Source 
* Expose AdamW parameters

* Use reference
This commit is contained in:
Dave Lage
2023-12-16 19:41:56 -05:00
committed by GitHub
parent c4cfcf1539
commit c630622a07
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-nn/src/optim.rs
View File

@ -190,4 +190,12 @@ impl AdamW {
}; };
Self::new(vars, params) Self::new(vars, params)
} }
pub fn params(&self) -> &ParamsAdamW {
&self.params
}
pub fn set_params(&mut self, params: ParamsAdamW) {
self.params = params;
}
} }