epoch = 10.0
train_loss = 0.10352115373611451
train_runtime = 364.2147
train_samples_per_second = 27.456
train_steps_per_second = 6.864