diff options
Diffstat (limited to 'candle-nn/src/lib.rs')
| -rw-r--r-- | candle-nn/src/lib.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-nn/src/lib.rs b/candle-nn/src/lib.rs index db01b067..d0b62dbb 100644 --- a/candle-nn/src/lib.rs +++ b/candle-nn/src/lib.rs @@ -15,6 +15,7 @@ pub mod vision; pub use activation::Activation; pub use conv::{Conv1d, Conv1dConfig}; pub use embedding::Embedding; +pub use init::Init; pub use layer_norm::LayerNorm; pub use linear::Linear; pub use optim::SGD; |