- Status has entity paarmeter
- train provides entity parameter to Status
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user