Fix missing callback refactoring.
Fix missing callback refactoring.

DeepGnosis/keras,kuza55/keras,kemaswill/keras,relh/keras,nebw/keras,keras-team/keras,dolaameng/keras,keras-team/keras,daviddiazvico/keras