epoch = 0.3
train_runtime = 2847.5807
train_samples_per_second = 1.692