Python-сообщество

Хотел спросить - как максимально удобно получать статитстику, за каждую эпоху обучения в Keras?
МОжно ли как нибудь в нег например запихнуть каллбак который будет вызываться после окончания одной эпохи?