File size: 249 Bytes
e0db531
 
 
 
 
 
 
1
2
3
4
5
6
7
8
#!/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