diff options
Diffstat (limited to 'candle-nn/tests/optim.rs')
| -rw-r--r-- | candle-nn/tests/optim.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/tests/optim.rs b/candle-nn/tests/optim.rs index 1327ae91..899745d4 100644 --- a/candle-nn/tests/optim.rs +++ b/candle-nn/tests/optim.rs @@ -6,7 +6,7 @@ use test_utils::{to_vec0_round, to_vec2_round}; use anyhow::Result; use candle::{Device, Tensor, Var}; -use candle_nn::{AdamW, Linear, ParamsAdamW, SGD}; +use candle_nn::{AdamW, Linear, Module, ParamsAdamW, SGD}; #[test] fn sgd_optim() -> Result<()> { |