diff options
Diffstat (limited to 'candle-nn/src')
| -rw-r--r-- | candle-nn/src/optim.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/src/optim.rs b/candle-nn/src/optim.rs index 741c51dc..a8b5b370 100644 --- a/candle-nn/src/optim.rs +++ b/candle-nn/src/optim.rs @@ -39,7 +39,7 @@ impl SGD { let grads = loss.backward()?; for var in self.vars.iter() { if let Some(grad) = grads.get(var) { - var.set(&var.sub(&(grad * self.learning_rate)?)?)? + var.set(&var.sub(&(grad * self.learning_rate)?)?)?; } } Ok(()) |