run_det_only_coordfix.sh

#!/bin/bash
# Detection-only baseline with coordinate system fix.
# No planning data, no resume — trains from scratch on detection only.
cd /home/guoyuanbo/3dtokenizer-atlas

ENV_ROOT=/home/guoyuanbo/3dtokenizer-atlas/envs/streampetr
export PATH="${ENV_ROOT}/bin:${PATH}"
export LD_LIBRARY_PATH="${ENV_ROOT}/lib:${LD_LIBRARY_PATH:-}"

exec ${ENV_ROOT}/bin/deepspeed --num_gpus=4 train_atlas.py \
    --llm_model pretrained/vicuna-7b-v1.5 \
    --data_json data/atlas_nuscenes_train.json \
    --data_root /home/guoyuanbo/autodl-tmp/data/nuscenes \
    --visual_token_mode online \
    --task_balance_mode none \
    --task_loss_weights detection=1.0 \
    --planning_table3_mode atlas_base \
    --streampetr_config configs/streampetr_atlas_aligned.py \
    --streampetr_ckpt pretrained/streampetr/streampetr_eva02_ep24.pth \
    --topomlp_config configs/topomlp_atlas_aligned.py \
    --topomlp_ckpt work_dirs/topomlp_atlas_aligned/epoch_24.pth \
    --deepspeed configs/ds_zero2.json \
    --output_dir work_dirs/atlas_det_only_coordfix \
    --epochs 3 \
    --lr 2e-5 \
    --weight_decay 1e-4 \
    --batch_size 1 \
    --gradient_accumulation_steps 2 \
    --warmup_ratio 0.03 \
    --max_grad_norm 1.0 \
    --log_steps 100 \
    --save_epochs 1 \
    --keep_last_n_ckpts 3 \
    --seed 42 \
    --num_workers 4