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 8ab51070..e9552e83 100644 --- a/candle-nn/src/lib.rs +++ b/candle-nn/src/lib.rs @@ -23,6 +23,7 @@ pub use group_norm::{group_norm, GroupNorm}; pub use init::Init; pub use layer_norm::{layer_norm, rms_norm, LayerNorm, LayerNormConfig, RmsNorm}; pub use linear::{linear, linear_no_bias, Linear}; +pub use ops::Dropout; pub use optim::{AdamW, ParamsAdamW, SGD}; pub use rnn::{lstm, LSTM, RNN}; pub use var_builder::VarBuilder; |