batch_size: 128
epochs: 10
learning_rate: 0.001
data_dir: "./data"
logs_dir: "./logs"