File size: 770 Bytes
1d6f391 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | #!/bin/bash
#SBATCH --job-name=prb_v5f_c
#SBATCH --partition=nova
#SBATCH --gres=gpu:a100:1
#SBATCH --cpus-per-task=8
#SBATCH --mem=128G
#SBATCH --time=04:00:00
#SBATCH --output=bert_v6_contrastive/scripts/probes_v5f_cached_%j.out
#SBATCH --error=bert_v6_contrastive/scripts/probes_v5f_cached_%j.err
echo "V5a FULL cached probes | Job $SLURM_JOB_ID | $(hostname) | $(date)"
source /work/ratul1/supantha/miniconda3/etc/profile.d/conda.sh
conda activate blade
python3 -u bert_v6_contrastive/scripts/novel_probes_v2.py \
--model v5 --max_samples 999999 \
--probe composition polymerization size_prediction mlm_zeroshot token_importance ambiguity \
--output_dir bert_v6_contrastive/analysis/novel_probes_v2_expanded_v5 \
--device cuda
echo "Done $(date)"
|