Skip to content

Commit

Permalink
Merge branch 'main' into conv1d
Browse files Browse the repository at this point in the history
  • Loading branch information
coreylowman committed Jan 25, 2024
2 parents 89952a1 + a4a271f commit 0e3d1ed
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dfdx/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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` |
//!
Expand Down

0 comments on commit 0e3d1ed

Please sign in to comment.