From 74138b5514608e2738e6edbfd937a6c50f534124 Mon Sep 17 00:00:00 2001
From: Jens Ahrensfeld <jens@jayfield.org>
Date: Mon, 5 Jan 2026 20:53:31 +0100
Subject: [PATCH] train: swapped epochs and mini-batch loop

---
 src/rbm/train.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/rbm/train.py b/src/rbm/train.py
index b166b7c..38fca06 100644
--- a/src/rbm/train.py
+++ b/src/rbm/train.py
@@ -214,12 +214,13 @@ def train(entity: Entity, batch: Mat, status: Status):
 	if entity.type == Entity.Type.BG_RBM:
 		cd_func = cd_binary_gaussian
 
-	for mini_batch in to_mini_batch(batch, mini_batch_size):
+	for epochs in range(params.num_epochs):
 		if not keep_running:
 			break
 
-		entity.grad_zero()
-		for epochs in range(params.num_epochs):
+		for mini_batch in to_mini_batch(batch, mini_batch_size):
+			# Reset gradient
+			entity.grad_zero()
 			# Contrastive divergence learning: calculate gradients
 			dwhv, dbv, dbh = cd_func(entity, mini_batch)