epoch = 3.0
train_runtime = 922.3702
train_samples_per_second = 2.28