epoch = 0.98
train_runtime = 82414.2278
train_samples_per_second = 1.213