- changed status interval
- show number of remaining epochs at beginning of training
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
+1
-1
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user