From 7ee48ac60dd7fc99122f90e859ec1401da41c6ff Mon Sep 17 00:00:00 2001
From: Felix Andreas <felix.andreas@melli.com>
Date: Wed, 22 Mar 2023 23:48:12 +0100
Subject: [PATCH] fix: Actually use different batches during training

Co-authored-by: Philipp <kjell-philipp_hahn@web.de>
---
 src/lib.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/lib.c b/src/lib.c
index 974503e..d7793af 100644
--- a/src/lib.c
+++ b/src/lib.c
@@ -310,7 +310,7 @@ void epoch(Network network, Dataset dataset, int batch_size, double learning_rat
     printf("Start epoch with %d batches (batch_size: %d)\n", batches, batch_size);
     for (int i = 0; i < batches; i++)
     {
-        double loss = update_mini_batch(network, dataset.images, batch_size, learning_rate) / batch_size;
+        double loss = update_mini_batch(network, dataset.images + i * batch_size, batch_size, learning_rate) / batch_size;
         printf("loss: %.4f\r", loss);
     }
     printf("\n");