models/bert/velocity_z/train_results.json

{
    "epoch": 12.0,
    "train_loss": 0.5175488976871266,
    "train_runtime": 188.9864,
    "train_samples": 8689,
    "train_samples_per_second": 551.722,
    "train_steps_per_second": 4.318
}