Skip to content

Commit 097e468

Browse files
committed
Callback robustness fix
1 parent b5f65df commit 097e468

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

keras/callbacks.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@ def on_batch_begin(self, batch, logs={}):
4949
self._t_enter_batch = time.time()
5050

5151
def on_batch_end(self, batch, logs={}):
52+
if not hasattr(self, '_t_enter_batch'):
53+
self._t_enter_batch = time.time()
5254
self._delta_t_batch = time.time() - self._t_enter_batch
5355
t_before_callbacks = time.time()
5456
for callback in self.callbacks:

0 commit comments

Comments
 (0)