data_dir: data/processed
epochs: 5
batch_size: 256
learning_rate: 5e-05
epoch: 4