epoch = 5.0
train_loss = 2.181030597859639
train_runtime = 4138.7205
train_samples = 42367
train_samples_per_second = 51.184
train_steps_per_second = 0.8