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

This commit is contained in:
Matt Graham 2016-10-17 23:28:01 +01:00
parent 0cec9920f3
commit f13ba91f82
2 changed files with 1976 additions and 22 deletions

View File

@ -124,7 +124,8 @@ class Optimiser(object):
recorded to their column index in the array and the final value recorded to their column index in the array and the final value
being the total time elapsed in seconds during the training run. 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() run_start_time = time.time()
for epoch in range(1, num_epochs + 1): for epoch in range(1, num_epochs + 1):
for scheduler in self.schedulers: for scheduler in self.schedulers:

File diff suppressed because one or more lines are too long