epoch = 20.0
train_loss = 0.4021474394976566
train_runtime = 51134.1112
train_samples_per_second = 11365.472
train_steps_per_second = 44.396