diff options
Diffstat (limited to 'candle-nn/examples/basic_optimizer.rs')
| -rw-r--r-- | candle-nn/examples/basic_optimizer.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/examples/basic_optimizer.rs b/candle-nn/examples/basic_optimizer.rs index 3c5665e8..cd5824dd 100644 --- a/candle-nn/examples/basic_optimizer.rs +++ b/candle-nn/examples/basic_optimizer.rs @@ -1,5 +1,5 @@ use candle::{DType, Device, Result, Tensor}; -use candle_nn::{linear, AdamW, Linear, ParamsAdamW, VarBuilder, VarMap}; +use candle_nn::{linear, AdamW, Linear, Module, ParamsAdamW, VarBuilder, VarMap}; fn gen_data() -> Result<(Tensor, Tensor)> { // Generate some sample linear data. |