- Status has entity paarmeter

- train provides entity parameter to Status
This commit is contained in:
2026-01-06 11:29:47 +01:00
parent 42520e5761
commit ff6254a15d
3 changed files with 13 additions and 10 deletions
+1
View File
@@ -76,6 +76,7 @@ class Entity:
self.enable_training = enable_training
self.state = RbmState.from_layer_params(shape)
self.grad = RbmState.from_layer_params(shape)
self.name = f"Entity-{shape[0]}x{shape[1]}"
self.type = None
if params.do_gaussian_visible:
if params.do_gaussian_hidden: