epoch = 3.0
train_loss = 0.3180885013734469
train_runtime = 6829.1181
train_samples_per_second = 121.07
train_steps_per_second = 0.946