epoch = 100.0
train_loss = 0.43426066813115094
train_runtime = 194605.8141
train_samples_per_second = 814.294
train_steps_per_second = 0.582