key_commit: tr training WIP