From e8624e1ebcabcc831d651e0beefe32df1463c903 Mon Sep 17 00:00:00 2001
From: jvech <jmvalenciae@unal.edu.co>
Date: Mon, 26 Aug 2024 12:12:40 -0500
Subject: add: mini batch learning implemented

---
 src/main.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/main.c')

diff --git a/src/main.c b/src/main.c
index 22737dc..3dc1b5f 100644
--- a/src/main.c
+++ b/src/main.c
@@ -91,6 +91,7 @@ int main(int argc, char *argv[]) {
     char default_config_path[512], *env_config_path;
     struct Configs ml_configs = {
         .epochs = 100,
+        .batch_size = 32,
         .alpha = 1e-5,
         .config_filepath = "",
         .network_size = 0,
@@ -138,6 +139,7 @@ int main(int argc, char *argv[]) {
                 y.data, y.shape,
                 load_loss(ml_configs),
                 ml_configs.epochs,
+                ml_configs.batch_size,
                 ml_configs.alpha);
         nn_network_write_weights(ml_configs.weights_filepath, network, ml_configs.network_size);
         fprintf(stderr, "weights saved on '%s'\n", ml_configs.weights_filepath);
-- 
cgit v1.2.3-70-g09d2