flow_grpo_cxr / scripts /single_node /server_profile_defaults.sh
zhui711's picture
Upload folder using huggingface_hub
535fb25 verified
Raw
History Blame Contribute Delete
1.19 kB
#!/usr/bin/env bash
apply_server_profile_defaults() {
case "${SERVER_PROFILE:-}" in
"")
;;
h20_ceph)
local h20_base="/apdcephfs_zwfy2/share_303944931/connorxian/Code/medUni/wt/zr"
export FLOW_GRPO_ROOT="${FLOW_GRPO_ROOT:-${h20_base}/flow_grpo_cxr}"
export OMNIGEN_CODE_ROOT="${OMNIGEN_CODE_ROOT:-${h20_base}/gen_joint}"
export SFT_LORA_PATH="${SFT_LORA_PATH:-${h20_base}/gen_joint/results_new/scratch_15k}"
export DATASET_ROOT="${DATASET_ROOT:-${h20_base}/dataset}"
export TRAIN_JSONL="${TRAIN_JSONL:-${h20_base}/dataset/anno/cxr_synth_anno_mask_train.jsonl}"
export TEST_JSONL="${TEST_JSONL:-${h20_base}/dataset/anno/cxr_synth_anno_mask_test.jsonl}"
export DATASET_PATH_REMAP_FROM="${DATASET_PATH_REMAP_FROM:-/home/wenting/dataset/LIDC_IDRI}"
export DATASET_PATH_REMAP_TO="${DATASET_PATH_REMAP_TO:-${DATASET_ROOT}}"
export MIRP_REPO_ROOT="${MIRP_REPO_ROOT:-${h20_base}/mirp}"
export HF_HOME="${HF_HOME:-${HOME}/.cache/huggingface}"
export HF_HUB_CACHE="${HF_HUB_CACHE:-${HF_HOME}/hub}"
;;
*)
echo "Unsupported SERVER_PROFILE: ${SERVER_PROFILE}" >&2
return 1
;;
esac
}