forked from snap-stanford/GraphGym
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_batch_pyg.sh
25 lines (22 loc) · 990 Bytes
/
run_batch_pyg.sh
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
#!/usr/bin/env bash
CONFIG=example_node
GRID=example
REPEAT=3
MAX_JOBS=8
SLEEP=1
MAIN=main_pyg
# generate configs (after controlling computational budget)
# please remove --config_budget, if don't control computational budget
python configs_gen.py --config configs/pyg/${CONFIG}.yaml \
--grid grids/pyg/${GRID}.txt \
--out_dir configs
#python configs_gen.py --config configs/ChemKG/${CONFIG}.yaml --config_budget configs/ChemKG/${CONFIG}.yaml --grid grids/ChemKG/${GRID}.txt --out_dir configs
# run batch of configs
# Args: config_dir, num of repeats, max jobs running, sleep time
bash parallel.sh configs/${CONFIG}_grid_${GRID} $REPEAT $MAX_JOBS $SLEEP $MAIN
# rerun missed / stopped experiments
bash parallel.sh configs/${CONFIG}_grid_${GRID} $REPEAT $MAX_JOBS $SLEEP $MAIN
# rerun missed / stopped experiments
bash parallel.sh configs/${CONFIG}_grid_${GRID} $REPEAT $MAX_JOBS $SLEEP $MAIN
# aggregate results for the batch
python agg_batch.py --dir results/${CONFIG}_grid_${GRID}