diff --git a/dfdx/src/lib.rs b/dfdx/src/lib.rs index b07b77df..235ca50f 100644 --- a/dfdx/src/lib.rs +++ b/dfdx/src/lib.rs @@ -224,7 +224,7 @@ //! | Optimizer | dfdx | pytorch | //! | --- | --- | --- | //! | SGD | [nn::optim::Sgd] | `torch.optim.SGD` | -//! | Adam | [nn::optim::Adam] | torch.optim.Adam` | +//! | Adam | [nn::optim::Adam] | `torch.optim.Adam` | //! | AdamW | [nn::optim::Adam] with [nn::optim::WeightDecay::Decoupled] | `torch.optim.AdamW` | //! | RMSprop | [nn::optim::RMSprop] | `torch.optim.RMSprop` | //!