my-bert-model/train_results.json

{
    "epoch": 2.0,
    "train_loss": 0.3888542595810778,
    "train_runtime": 268.9983,
    "train_samples_per_second": 45.279,
    "train_steps_per_second": 2.833
}