official/nlp/bert/model_training_utils.py · 2d05fc8af6533babd8b5ebc44149758c6b2535e7 · ModelZoo / ResNet50_tensorflow

Add calling of custom callback on_epoch_begin() and on_epoch_end() methods. · 31da2245

Jeremiah Harmsen authored Apr 16, 2020

Minor refactoring to (1) encapsulate callbacks in a CallbackList and (2) allow evaluation results to be threaded to on_epoch_end callback.

PiperOrigin-RevId: 306827406

31da2245

model_training_utils.py 20.4 KB

Replace model_training_utils.py