epoch = 3.0
train_loss = 0.3398423703541086
train_runtime = 2416.9734
train_samples_per_second = 342.082
train_steps_per_second = 2.672