epoch = 1.0
train_loss = 0.6395634657499806
train_runtime = 1631.1404
train_samples = 48328
train_samples_per_second = 29.628
train_steps_per_second = 0.463