epoch = 5.0
train_loss = 1.8752608189770101
train_runtime = 3750.4732
train_samples = 42367
train_samples_per_second = 56.482
train_steps_per_second = 0.883