Dealing with case where number of epochs less than stats interval.
This commit is contained in:
parent
0cec9920f3
commit
f13ba91f82
@ -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
Loading…
Reference in New Issue
Block a user