epoch = 4.0
train_loss = 2.831424217569864
train_runtime = 8708.1966
train_samples = 42367
train_samples_per_second = 19.461
train_steps_per_second = 1.216