SACFlow / scripts /amos /eval_all_main.sh
sathiiii's picture
Add scripts
e7dc055 verified
Raw
History Blame Contribute Delete
822 Bytes
#!/usr/bin/env bash
set -euo pipefail
declare -A CFGS
CFGS[source_ct]=configs/experiments/amos_source_ct_shared13.yaml
CFGS[oracle_mri]=configs/experiments/amos_oracle_mri_shared13.yaml
CFGS[selftrain]=configs/experiments/amos_selftrain_ct2mri_shared13.yaml
CFGS[peft]=configs/experiments/amos_peft_ct2mri_shared13.yaml
CFGS[proto_align]=configs/experiments/amos_proto_align_ct2mri_shared13.yaml
CFGS[sacflow]=configs/experiments/amos_sacflow_ct2mri_shared13.yaml
for name in source_ct oracle_mri selftrain peft proto_align sacflow; do
outdir=$(python - <<PY
from sacflow.utils.config import load_yaml
print(load_yaml('${CFGS[$name]}')['output_dir'])
PY
)
echo "=== Evaluating $name from $outdir ==="
python tools/eval.py --config "${CFGS[$name]}" --checkpoint "$outdir/checkpoints/best.pt" --split target_val
done