-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathconfig_cross_validate.yaml
34 lines (27 loc) · 1.03 KB
/
config_cross_validate.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# cross validation config for regression/classification
defaults:
# data
- datamodule: classification/schneider.yaml
# model
- model/encoder: encoder.yaml
- model/decoder: classifier.yaml
# training
- optimizer: adams.yaml
# lightning settings
- trainer: minimal.yaml
- logger: wandb.yaml
- callbacks: predictive.yaml
# cv
- cross_validate: kfold.yaml
# other configs
seed: 35
skip_test: False # skip running model against test set after training finishes
git_repo_path: null # path to the git repo (used to log commit)
# git_repo_path: /Users/mjwen/Applications/rxnrep
return_val_metric_score: False # let the train function return val metric instead of test metric
# path to original working directory
# hydra hijacks working directory by changing it to the current log directory,
# something like: .../outputs/2021-05-03/11-26-06
# It's useful to have original working directory ...
# learn more here: https://hydra.cc/docs/next/tutorials/basic/running_your_app/working_directory
original_working_dir: ${hydra:runtime.cwd}