| # 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 | |