Skip to content

Commit

Permalink
bugfix, batch_bar doesn't exist if train started after all epochs com…
Browse files Browse the repository at this point in the history
…pleted.
  • Loading branch information
firatozdemir committed Sep 11, 2023
1 parent 4c956c2 commit 03c362a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tqdm/keras.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ def on_epoch_begin(self, epoch, *_, **__):
raise KeyError('Unknown verbosity')

def on_train_end(self, *_, **__):
if self.verbose:
if hasattr(self, 'batch_bar'):
self.batch_bar.close()
self.epoch_bar.close()

Expand Down

0 comments on commit 03c362a

Please sign in to comment.