Expose AdamW parameters (huggingface#1449)

* Expose AdamW parameters * Use reference
wonkyDD · Dec 17, 2023 · c630622 · c630622
1 parent c4cfcf1
commit c630622
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/candle-nn/src/optim.rs b/candle-nn/src/optim.rs
@@ -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;
+    }
 }