12 lines
231 B
YAML
12 lines
231 B
YAML
prepare:
|
|
data_dir: data/raw
|
|
save_dir: data/processed
|
|
train_valid_split: [0.7, 0.3]
|
|
random_seed: 0
|
|
train:
|
|
data_dir: data/processed
|
|
epochs: 5
|
|
batch_size: 256
|
|
learning_rate: 0.00005
|
|
evaluate:
|
|
data_dir: data/processed |