-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_all_experiments.sh
19 lines (16 loc) · 1.22 KB
/
run_all_experiments.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
################################################
# Train the models #
################################################
# Optionally, set CUDA_VISIBLE_DEVICES and pass --neptune-logger
python tsl_experiments.py --dataset-name la --model-name fcrnn --config fcrnn.yaml
python tsl_experiments.py --dataset-name la --model-name dcrnn --config dcrnn.yaml
python tsl_experiments.py --dataset-name la --model-name gwnet --config gwnet.yaml
python tsl_experiments.py --dataset-name la --model-name gatedgn --config gatedgn.yaml
python tsl_experiments.py --dataset-name bay --model-name fcrnn --config fcrnn.yaml
python tsl_experiments.py --dataset-name bay --model-name dcrnn --config dcrnn.yaml
python tsl_experiments.py --dataset-name bay --model-name gwnet --config gwnet.yaml
python tsl_experiments.py --dataset-name bay --model-name gatedgn --config gatedgn.yaml
python tsl_experiments.py --dataset-name gpolyvar --model-name fcrnn --config fcrnn.yaml
python tsl_experiments.py --dataset-name gpolyvar --model-name dcrnn --config dcrnn.yaml
python tsl_experiments.py --dataset-name gpolyvar --model-name gwnet --config gwnet.yaml
python tsl_experiments.py --dataset-name gpolyvar --model-name gatedgn --config gatedgn.yaml