From ec5d259ff7762ee2dfb2a3e81a406c7b3d9b562f Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Tue, 6 Jan 2026 09:44:29 +0100 Subject: [PATCH] Entity: added property enable training --- src/rbm/entity.py | 3 ++- src/rbm/model.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/rbm/entity.py b/src/rbm/entity.py index 68c6f74..6c3ec64 100644 --- a/src/rbm/entity.py +++ b/src/rbm/entity.py @@ -69,10 +69,11 @@ class Entity: GB_RBM = "GB-RBM" GG_RBM = "GG-RBM" - def __init__(self, shape: tuple[int, int], params: EntityParams, training_params: TrainingParams|None = None): + def __init__(self, shape: tuple[int, int], params: EntityParams, training_params: TrainingParams|None = None, enable_training: bool = True): self.shape = shape self.params = params self.training_params = training_params + self.enable_training = enable_training self.state = RbmState.from_layer_params(shape) self.grad = RbmState.from_layer_params(shape) self.type = None diff --git a/src/rbm/model.py b/src/rbm/model.py index bec0b25..a2416c0 100644 --- a/src/rbm/model.py +++ b/src/rbm/model.py @@ -24,8 +24,9 @@ class Model(ABC): entities = self.objects(Entity) _batch = np.copy(batch) for entity in entities: - if train(entity, _batch, Status()): - _batch = entity.forward(_batch) + if entity.enable_training: + train(entity, _batch, Status()) + _batch = entity.forward(_batch) def forward(self, x: Mat) -> Mat: pass