# EXP=predict2_video2world_training_2b_cosmos_nemo_assets
# EXP=debug
torchrun --nproc_per_node=8 --master_port=12341 -m scripts.train \
    --config=cosmos_predict2/configs/base/config.py -- \
    experiment=$1 $2