-
Notifications
You must be signed in to change notification settings - Fork 21
/
_base_data_config.yaml
20 lines (17 loc) · 1.04 KB
/
_base_data_config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# --> EDIT the data_dir to the path where the data is stored, if necessary:
# Make sure that ``data_dir`` points to an absolute path!
# Note: ${oc.env:HOME} is the path to home directory
# By default, the datamodules can automatically infer if the data is stored in the subdirectories
# $data_dir/oisstv2 or $data_dir/physical-nn-benchmark
# So, if you adhere to this naming, you can keep data_dir to point to their parent dir. By default, ``$HOME/data``
data_dir: "${oc.env:HOME}/data"
# --> Optional edits:
batch_size: 16 # batch size for training
eval_batch_size: ${datamodule.batch_size} # batch size for validation/testing (larger might be faster, but uses more RAM)
num_workers: 4 # number of workers for data loading (-1 means all available CPUs)
pin_memory: True
drop_last: False
verbose: ${verbose}
window: 1
horizon: 1 # number of time steps to predict ahead
# seed_data: 77 # random seed for reproducibility of random data split (different to the one for training)