epoch = 5.0
train_loss = 1.8827527066371952
train_runtime = 3344.9943
train_samples = 42367
train_samples_per_second = 63.329
train_steps_per_second = 1.979