Downloads: 1
Diff of
/AICare-baselines/configs/config.yaml
[000000]
..
[0f1df3]
Switch to unified view
| a | b/AICare-baselines/configs/config.yaml | ||
|---|---|---|---|
| 1 | stage: tune # tune / train (k-fold) / test |
||
| 2 | batch_size: 81920 # make sure the default value contains the whole dataset |
||
| 3 | epochs: 50 |
||
| 4 | patience: 10 |
||
| 5 | learning_rate: 1e-3 |
||
| 6 | output_dim: 1 |