From c78ce765016392673805ed8dfafb4ae1a7b6c26f Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Fri, 18 Aug 2023 09:38:22 +0100
Subject: Add a simple Module trait and implement it for the various nn layers
 (#500)

* Start adding the module trait.

* Use the module trait.

* Implement module for qmatmul.
---
 candle-nn/examples/basic_optimizer.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'candle-nn/examples/basic_optimizer.rs')

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.
-- 
cgit v1.2.3