Entity: added property enable training

2026-01-06 09:44:29 +01:00
parent b1fe7229fa
commit ec5d259ff7
2 changed files with 5 additions and 3 deletions
@@ -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