- training params are (again) attrubute of Entity

- ditched layout concept
2026-01-02 14:51:19 +01:00
parent 8b97350564
commit 10691894d1
8 changed files with 70 additions and 98 deletions
@@ -1,12 +1,10 @@
-from .state import RbmState
-from .train import TrainingParams
-from .entity import Entity, EntityParams
+from .entity import Entity, EntityParams, TrainingParams

 class Layer:
 	def __init__(self, name: str, shape: tuple[int, int, int, int], entity_params: EntityParams, training_params: TrainingParams):
 		self.name = name
 		self.shape = shape
-		self.entity = Entity((shape[0]*shape[1]+shape[2], shape[3]), entity_params)
+		self.entity = Entity((shape[0]*shape[1]+shape[2], shape[3]), entity_params, training_params)
 		self.training_params = training_params

 	def init(self, std: float):