- 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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user