-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_task_ablation.sh
12 lines (9 loc) · 1.03 KB
/
run_task_ablation.sh
1
2
3
4
5
6
7
8
9
10
11
12
# MFRNP-H
python train.py --data_path "data/full_dataset/poisson" --save_dir poisson --config pde_config_ablation.yaml --levels 3 --device cuda
python train.py --data_path "data/full_dataset/poisson" --save_dir poisson --config pde_config_ablation.yaml --levels 5 --device cuda
python train.py --data_path "data/OOD/poisson/l3" --save_dir poisson_OOD --config pde_config_ablation.yaml --levels 3 --device cuda
python train.py --data_path "data/OOD/poisson/l5" --save_dir poisson_OOD --config pde_config_ablation.yaml --levels 5 --device cuda
python train.py --data_path "data/full_dataset/heat" --save_dir heat --config pde_config_ablation.yaml --levels 3 --device cuda
python train.py --data_path "data/full_dataset/heat" --save_dir heat --config pde_config_ablation.yaml --levels 5 --device cuda
python train.py --data_path "data/OOD/heat/l3" --save_dir heat_OOD --config pde_config_ablation.yaml --levels 3 --device cuda
python train.py --data_path "data/OOD/heat/l5" --save_dir heat_OOD --config pde_config_ablation.yaml --levels 5 --device cuda