code/probes/cluster_scripts/run_probes_v5_res_external.sh

#!/bin/bash
#SBATCH --job-name=prb_v5r_e
#SBATCH --partition=nova
#SBATCH --gres=gpu:a100:1
#SBATCH --cpus-per-task=8
#SBATCH --mem=64G
#SBATCH --time=02:00:00
#SBATCH --output=bert_v6_contrastive/scripts/probes_v5r_ext_%j.out
#SBATCH --error=bert_v6_contrastive/scripts/probes_v5r_ext_%j.err

echo "V5a RESOLVED external 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 --resolved_only \
    --probe cancer_markers glycosylation_type taxonomic_class knn_purity link_binary \
    --output_dir bert_v6_contrastive/analysis/novel_probes_v2_res_expanded_v5 \
    --device cuda

echo "Done $(date)"