`Stage.current_epoch` is inconsistent

[sehoffmann]

Depends on callback priority AND what hook, e.g. pre_epoch vs post_epoch. This is not good. Should be consistent until all callbacks are done and the next epoch starts