jiarui_li/DeepHealth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add --max_cpu_cores argument for parallel processing in evaluation scripts

Browse Source
This commit is contained in:
Jiarui Li
2026-01-17 23:53:24 +08:00
parent 248fb09c34
commit cfe7f88162
4 changed files with 234 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
run_evaluations_multi_gpu.sh
View File

@@ -31,7 +31,7 @@ Options:
Common eval args:
Anything after `--` is appended to BOTH evaluation commands.
Use this only for flags supported by BOTH scripts (e.g. --batch_size, --num_workers, --seed, --min_pos, --no_tqdm).
Use this only for flags supported by BOTH scripts (e.g. --batch_size, --num_workers, --max_cpu_cores, --seed, --min_pos, --no_tqdm).
Per-eval args:
For eval-specific flags (e.g. evaluate_horizon.py --topk_list / --workload_fracs), use --horizon-args-file.
@@ -41,6 +41,8 @@ Examples:
./run_evaluations_multi_gpu.sh --gpus 0,1
./run_evaluations_multi_gpu.sh --gpus 0,1 --runs-root runs --pattern "delphi_*" \
--horizons 0.25 0.5 1 2 5 10 --age-bins 40 45 50 55 60 65 70 75 inf -- --batch_size 512 --num_workers 4
./run_evaluations_multi_gpu.sh --gpus 0,1 --runs-root runs --pattern "delphi_*" \
-- --batch_size 512 --num_workers 4 --max_cpu_cores -1
USAGE
}