epoch = 3.0
train_loss = 0.3292924272956854
train_runtime = 6840.1136
train_samples_per_second = 120.876
train_steps_per_second = 0.944