supanthadey1's picture
Add BERTose and AFFINose training code release
1d6f391 verified
Raw
History Blame Contribute Delete
929 Bytes
#!/bin/bash
#SBATCH --job-name=probes_res_v6
#SBATCH --partition=nova
#SBATCH --gres=gpu:a100:1
#SBATCH --cpus-per-task=8
#SBATCH --mem=64G
#SBATCH --time=04:00:00
#SBATCH --output=bert_v6_contrastive/scripts/probes_resolved_v6_%j.out
#SBATCH --error=bert_v6_contrastive/scripts/probes_resolved_v6_%j.err
echo "Resolved-only probes — V6 | Job $SLURM_JOB_ID | $(hostname) | $(date)"
source /work/ratul1/supantha/miniconda3/etc/profile.d/conda.sh
conda activate blade
cd /work/ratul1/supantha/glycan-SD-VS/bert_training_v3/v3.1_cluster_training
pip install -q scikit-learn umap-learn 2>/dev/null
python3 -u bert_v6_contrastive/scripts/novel_probes_v2.py \
--model v6 --probe all --max_samples 999999 --resolved_only \
--output_dir bert_v6_contrastive/analysis/novel_probes_v2_resolved_v6 \
--device cuda
echo "Done V6 resolved: $(date)"
ls -la bert_v6_contrastive/analysis/novel_probes_v2_resolved_v6/ 2>/dev/null