- state load/save revised
- num gibbs samples is also part of entity - working 3-layer deep test model
This commit is contained in:
+2
-5
@@ -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)
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user