epoch = 1.0
train_runtime = 3834.5201
train_samples_per_second = 4.075