I am using callbacks (with default setting) in order to view training progress on Tensorboard. My Jupiter notebook training progress bar freezes after a certain amount of images