epoch = 5.0
train_loss = 1.8353108276413286
train_runtime = 3683.8727
train_samples = 42367
train_samples_per_second = 57.503
train_steps_per_second = 3.594