| #!/usr/bin/env bash |
| set -euo pipefail |
| CFG=${CFG:-configs/experiments/amos_source_ct_shared13.yaml} |
| TMP_CFG=${TMP_CFG:-configs/experiments/amos_source_ct_shared13_smoke.yaml} |
| python - <<'PY' |
| from sacflow.utils.config import load_yaml, save_yaml |
| cfg = load_yaml('configs/experiments/amos_source_ct_shared13.yaml') |
| cfg['output_dir'] = 'outputs/amos/shared13/source_ct_smoke' |
| cfg['train']['epochs'] = 1 |
| cfg['train']['steps_per_epoch'] = 2 |
| cfg['train']['val_every'] = 1 |
| cfg['train']['save_every'] = 1 |
| cfg['wandb']['enabled'] = False |
| save_yaml(cfg, 'configs/experiments/amos_source_ct_shared13_smoke.yaml') |
| PY |
| NPROC=${NPROC:-1} |
| torchrun --standalone --nnodes=1 --nproc_per_node=$NPROC tools/train.py --config "$TMP_CFG" --name amos_source_ct_smoke |
|
|
