Backup Save

include/neural_network/model/model.hpp

@@ -323,7 +323,7 @@ struct model{
      *  
      *
      */
-    bool add_optimizer_sgd(const panic::types::real_t learning_rate = static_cast<panic::types::real_t>(1e-3));
+    bool add_optimizer_sgd(const panic::types::real_t learning_rate = static_cast<panic::types::real_t>(1));
 
 
 
@@ -337,6 +337,22 @@ struct model{
      */
     bool finalize();
 
+
+    /**
+     * @brief Optimizes the model with parameters on trainable layers
+     *
+     * Computes:
+     * @code
+     * model.optimize()
+     * @endcode
+     *
+     * @return true if optimization is done correctly.
+     *
+     */
+    bool optimize();
+
+
+
     /**
      * @brief Trains the model with input data
      *