keras: fix resuming training when verbose>=2 (tqdm#1508)

mocitanefelibata · Feb 5, 2024 · 9a11b2e · 9a11b2e
1 parent 4c956c2
commit 9a11b2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tqdm/keras.py b/tqdm/keras.py
@@ -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()