jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

train: swapped epochs and mini-batch loop

Browse Source
This commit is contained in:
jens
2026-01-05 20:53:31 +01:00
parent 562e9e9bda
commit 74138b5514
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/train.py
+4 -3
View File
@@ -214,12 +214,13 @@ def train(entity: Entity, batch: Mat, status: Status):
if entity.type == Entity.Type.BG_RBM: if entity.type == Entity.Type.BG_RBM:
cd_func = cd_binary_gaussian 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: if not keep_running:
break break
entity.grad_zero() for mini_batch in to_mini_batch(batch, mini_batch_size):
for epochs in range(params.num_epochs): # Reset gradient
entity.grad_zero()
# Contrastive divergence learning: calculate gradients # Contrastive divergence learning: calculate gradients
dwhv, dbv, dbh = cd_func(entity, mini_batch) dwhv, dbv, dbh = cd_func(entity, mini_batch)