File size: 915 Bytes
535fb25 | 1 2 3 4 5 6 7 8 9 10 11 12 | # 1 GPU
# accelerate launch --config_file scripts/accelerate_configs/multi_gpu.yaml --num_processes=1 --main_process_port 29501 scripts/train_omnigen.py --config config/grpo.py:general_radiomics_omnigen_full_1gpu
export PYTHONPATH=/NAS_REMOTE/vicky/wt/codes/cxr_syn:/NAS_REMOTE/vicky/wt/codes/cxr_syn/flow_grpo:$PYTHONPATH
export OMNIGEN_CODE_ROOT=/NAS_REMOTE/vicky/wt/codes/cxr_syn/gen_code
export PYTHONPATH=$OMNIGEN_CODE_ROOT:$PYTHONPATH
export HF_TOKEN="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"
export HF_HOME="/NAS_REMOTE/vicky/wt/huggingface/models"
export HF_HUB_CACHE="/tmp/flow_grpo_hf_cache/hub"
export WANDB_API_KEY="e17ae3f7c8533a627b504b2f3a84b90d44fa3b31"
CUDA_VISIBLE_DEVICES=5 python3 -m accelerate.commands.launch --config_file scripts/accelerate_configs/multi_gpu.yaml --num_processes=1 --main_process_port 29501 scripts/train_omnigen.py --config config/grpo.py:general_radiomics_omnigen_full_1gpu
|