6 lines
86 B
YAML
6 lines
86 B
YAML
batch_size: 128
|
|
epochs: 10
|
|
learning_rate: 0.001
|
|
data_dir: "./data"
|
|
logs_dir: "./logs"
|