epoch = 3.0
train_loss = 0.3241952664897545
train_runtime = 7201.1171
train_samples_per_second = 114.816
train_steps_per_second = 0.897