stage: tune # tune / train (k-fold) / test
batch_size: 81920 # make sure the default value contains the whole dataset
epochs: 50
patience: 10
learning_rate: 1e-3
output_dim: 1
