Muti3d

Portable training package for the Muti3d / Gamma-World stage-2 experiment.

Quick start on a new machine:

cd Muti3d/train
./setup_env.sh
DOWNLOAD_CHECKPOINTS=1 ./setup_env.sh
# edit train/wandb.env.example and set WANDB_API_KEY
./stage2_singlenode.sh

Expected local files:

Muti3d/
  datasets/
    2p_3d/
      raw/...                     # Hugging Face clone layout
  data/
    raw/...                       # also supported local layout
  assets/
    checkpoints/
      Gamma-World/causal/model.safetensors
      Gamma-World/tokenizer.pth
      Cosmos-Reason1-7B/
  train/
    checkpoints/
      converted_causal_dcp/model/

Outputs are written under:

Muti3d/train/runs/

More details are in train/README_RUN.md.