supanthadey1's picture
Add BERTose and AFFINose training code release
1d6f391 verified
Raw
History Blame Contribute Delete
996 Bytes
#!/bin/bash
#SBATCH --job-name=probe15h_phylo
#SBATCH --output=logs/probe15h_phylo_%j.out
#SBATCH --error=logs/probe15h_phylo_%j.out
#SBATCH --partition=nova
#SBATCH --gres=gpu:1
#SBATCH --cpus-per-task=8
#SBATCH --mem=64G
#SBATCH --time=02:00:00
#SBATCH --mail-type=END,FAIL
#SBATCH --mail-user=supantha@iastate.edu
echo "============================================"
echo "Probe 15H: Phylogenetic Embedding"
echo "============================================"
echo "Started: $(date)"
echo "Node: $(hostname)"
echo "GPU: $(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null || echo 'N/A')"
module load miniconda3
conda activate glycanml
export PYTHONUNBUFFERED=1
cd /work/ratul1/supantha/glycan-SD-VS/bert_training_v3/v3.1_cluster_training
python bert_v6_contrastive/scripts/probe_15h_phylogenetic.py \
--model v6 \
--min-glycans 10
echo ""
echo "============================================"
echo "Completed: $(date)"
echo "============================================"