epoch = 5.0
train_loss = 2.08123623193931
train_runtime = 3339.4615
train_samples = 42367
train_samples_per_second = 63.434
train_steps_per_second = 1.982