From 32882657ca2299f114e18d831be56239ce5895c0 Mon Sep 17 00:00:00 2001 From: Eric Cardozo <139732027+xEricCardozo@users.noreply.github.com> Date: Mon, 16 Oct 2023 16:19:00 -0300 Subject: [PATCH] Update README.md --- README.md | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/README.md b/README.md index 0506d96..d26261a 100644 --- a/README.md +++ b/README.md @@ -77,8 +77,6 @@ struct Network : public net::Model { } net::optimizer::SGD optimizer {/*learning rate*/ 0.1}; - - void step() { optimizer.step(); } }; int main() { @@ -94,7 +92,7 @@ int main() { std::cout << loss_function.loss() << std::endl; loss_function.backward(); // backpropagate the gradients - network.step() // triggers the optimizer. + network.optimizer.step() // triggers the optimizer. } ```