#!/bin/bash
# Shorthand for running Any4D experiments.
# Usage: bash custom/run.sh experiment=any4d_foo [hydra overrides...]
# Equivalent to: python -m torch.distributed.run ... -m scripts.train --config=cosmos_predict2/configs/base/config.py -- $@

NGPU=${NGPU:-$(nvidia-smi -L 2>/dev/null | wc -l)}
PORT=${PORT:-12341}

exec torchrun --nproc_per_node="$NGPU" --master_port="$PORT" \
    -m scripts.train \
    --config=cosmos_predict2/configs/base/config.py \
    -- "$@"