epoch = 5.0
train_loss = 2.0527755599007507
train_runtime = 3751.6871
train_samples = 42367
train_samples_per_second = 56.464
train_steps_per_second = 0.882