Skip to content

Fix GPU selection and execution path handling#3

Open
Traveler-Roy wants to merge 1 commit into
CodeSoul-co:mainfrom
Traveler-Roy:codex/fix-gpu-selection
Open

Fix GPU selection and execution path handling#3
Traveler-Roy wants to merge 1 commit into
CodeSoul-co:mainfrom
Traveler-Roy:codex/fix-gpu-selection

Conversation

@Traveler-Roy

Copy link
Copy Markdown
  • Fix GPU selection defaults and priority:
    • default gpu_id now falls back to 0
    • explicit CLI --gpu overrides CUDA_VISIBLE_DEVICES
    • omitted --gpu preserves existing CUDA_VISIBLE_DEVICES
  • Add optional --gpu support across main.py, run_pipeline.py, and prepare_data.py.
  • Propagate explicit GPU selection from parent pipeline commands to child processes.
  • Update shell wrappers so they only forward GPU selection when the user explicitly provides it.
  • Harden DDP/rank-zero artifact handling for logging, evaluation, visualization, LoRA adapters, and BOW outputs.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant