Dealing with case where number of epochs less than stats interval.

2016-10-17 23:28:01 +01:00 · 2016-10-17 23:28:01 +01:00 · f13ba91f82
commit f13ba91f82
parent 0cec9920f3
2 changed files with 1976 additions and 22 deletions
--- a/mlp/optimisers.py
+++ b/mlp/optimisers.py
@ -124,7 +124,8 @@ class Optimiser(object):
            recorded to their column index in the array and the final value
            being the total time elapsed in seconds during the training run.
        """
-        run_stats = [self.get_epoch_stats().values()]
+        stats = self.get_epoch_stats()
+        run_stats = [stats.values()]
        run_start_time = time.time()
        for epoch in range(1, num_epochs + 1):
            for scheduler in self.schedulers:
--- a/notebooks/03_Multiple_layer_models.ipynb
+++ b/notebooks/03_Multiple_layer_models.ipynb