Skip to content

Support single or multi dataset path for training#535

Open
mingxueg-nv wants to merge 1 commit intoNVIDIA:mainfrom
mingxueg-nv:mingxue/training_with_multi_dataset_path
Open

Support single or multi dataset path for training#535
mingxueg-nv wants to merge 1 commit intoNVIDIA:mainfrom
mingxueg-nv:mingxue/training_with_multi_dataset_path

Conversation

@mingxueg-nv
Copy link
Copy Markdown

@mingxueg-nv mingxueg-nv commented Feb 5, 2026

Fixes #

Changes proposed in this pull request:

  • Support single or multi dataset path for training
torchrun --nproc_per_node=$NUM_GPUS --master_port=29500 \
    gr00t/experiment/launch_finetune.py \
    --base_model_path  nvidia/GR00T-N1.6-3B \
    --dataset_path /dataset1 /dataset2 /dataset3 ...  \
    --embodiment_tag NEW_EMBODIMENT \
     ....

Before submitting

  • I've read and followed all steps in the Making a pull request
    section of the CONTRIBUTING docs.
  • I've updated or added any relevant docstrings.
  • If this PR fixes a bug, I've added a test that will fail without my fix.
  • If this PR adds a new feature, I've added tests that sufficiently cover my new functionality.

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