VIPRA-reproduce / scripts /run_precompute.sh
yqi19's picture
VIPER reproduction: code, mini VIPER-19K pipeline, stage-1 checkpoint, qualitative comparisons and eval
e0db531 verified
Raw
History Blame Contribute Delete
249 Bytes
#!/bin/bash
cd "$(dirname "$0")/.."
source env.sh
export CUDA_VISIBLE_DEVICES=$((SLURM_LOCALID*2))
exec .venv/bin/python -m viper.precompute \
--clips data/annotated.jsonl --out_dir data/cache \
--shard $SLURM_LOCALID --num_shards $SLURM_NTASKS