diff --git a/src/rbm/stack_deep.py b/src/rbm/stack_deep.py index 62497e5..da28362 100644 --- a/src/rbm/stack_deep.py +++ b/src/rbm/stack_deep.py @@ -19,7 +19,7 @@ class StackDeep(Stack): def train(self, batch: np.ndarray, status=Status()): _batch = np.matrix.copy(batch) for index, layer in enumerate(self.layers): - print(f"Train layer {index}") + print(f"Train layer {index} for {layer.entity.params.num_epochs} epochs") _batch = self.batch_from(batch, index) layer.entity.train(_batch, cd_func=cd_jens, status=status) diff --git a/src/rbm/test_rbm.py b/src/rbm/test_rbm.py index 91a9890..3b41969 100644 --- a/src/rbm/test_rbm.py +++ b/src/rbm/test_rbm.py @@ -13,7 +13,7 @@ def cv_show(name: str, vec: np.array, shape): class MyStatus(Status): def __init__(self, _stack: StackDeep, _batch: np.ndarray): - Status.__init__(self, update_interval=1) + Status.__init__(self, update_interval=10) self.stack = _stack self.batch = _batch self.index = 0