Update README.md

mr-mapache · Oct 16, 2023 · 3288265 · 3288265
1 parent 8223506
commit 3288265
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/README.md b/README.md
@@ -77,8 +77,6 @@ struct Network : public net::Model<Network> {
     }
 
     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. 
 }
 
 ```