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
@@ -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