File size: 742 Bytes
e7dc055 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | #!/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
|