epoch = 5.0
train_loss = 2.181800748630912
train_runtime = 3900.3613
train_samples = 42696
train_samples_per_second = 54.733
train_steps_per_second = 0.856