impl Clone for Adam, SGD, RMSprop (coreylowman#775)

mattjurenka · May 13, 2023 · 6021491 · 6021491
1 parent ecfe402
commit 6021491
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/optim/adam/mod.rs b/src/optim/adam/mod.rs
@@ -70,7 +70,7 @@ impl Default for AdamConfig {
 /// ```
 ///
 /// See module level documentation at [crate::optim] for examples of how to actually use an optimizer.
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct Adam<M, E: Dtype, D: DeviceStorage> {
     /// Hyperparameter configuration
     pub cfg: AdamConfig,

diff --git a/src/optim/rmsprop/mod.rs b/src/optim/rmsprop/mod.rs
@@ -77,7 +77,7 @@ impl Default for RMSpropConfig {
 ///     centered: false,
 ///     weight_decay: Some(WeightDecay::Decoupled(1e-1)),
 /// });
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct RMSprop<M, E: Dtype, D: DeviceStorage> {
     /// Hyperparameter configuration
     pub cfg: RMSpropConfig,

diff --git a/src/optim/sgd/mod.rs b/src/optim/sgd/mod.rs
@@ -111,7 +111,7 @@ impl Default for SgdConfig {
 /// ```
 ///
 /// See module level documentation at [crate::optim] for examples of how to actually use an optimizer.
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub struct Sgd<M, E: Dtype, D: DeviceStorage> {
     /// Hyperparameter configuration
     pub cfg: SgdConfig,