lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Catch KeyboardInterrupt exception if the training is aborted

Browse Source
This commit is contained in:
Tristan Deleu 2015-05-28 11:14:24 +02:00
parent 2b1bb90bf0
commit dced184426
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
keras/models.py
View File

@ -239,6 +239,7 @@ class Sequential(Model):
for batch_index, (batch_start, batch_end) in enumerate(batches):
callbacks.on_batch_begin(batch_index)
try:
batch_ids = index_array[batch_start:batch_end]
X_batch = slice_X(X, batch_ids)
y_batch = y[batch_ids]
@ -249,6 +250,12 @@ class Sequential(Model):
else:
loss = self._train(*ins)
acc = None
except KeyboardInterrupt:
# If training is aborted, call the callbacks anyway before terminating
callbacks.on_batch_end(batch_index, [], 0., 0.)
callbacks.on_epoch_end(epoch, 0., 0.)
callbacks.on_train_end()
raise KeyboardInterrupt # TODO: Raise a more explicit Excpetion (?)
callbacks.on_batch_end(batch_index, batch_ids, loss, acc)