- state load/save revised

- num gibbs samples is also part of entity
- working 3-layer deep test model
This commit is contained in:
2025-12-21 17:34:11 +01:00
parent 22d189cf2f
commit 4bd5cffd6b
5 changed files with 57 additions and 30 deletions
+2 -5
View File
@@ -13,13 +13,10 @@ class Layer:
self.entity.state.init(mu=0, std=std)
def save(self, filename: str = None):
self.entity.state.to_file(filename)
self.entity.state.save(filename)
def load(self, filename: str = None):
state = RbmState.from_file(filename)
if state is not None:
self.entity.state = state
self.entity.state.load(filename)