supanthadey1's picture
Add BERTose and AFFINose training code release
1d6f391 verified
Raw
History Blame Contribute Delete
952 Bytes
#!/bin/bash
#SBATCH --job-name=probe15_taskD
#SBATCH --output=bert_v6_contrastive/cluster_scripts/slurm_logs/probe15d_%j.out
#SBATCH --error=bert_v6_contrastive/cluster_scripts/slurm_logs/probe15d_%j.err
#SBATCH --partition=nova
#SBATCH --gres=gpu:1
#SBATCH --cpus-per-task=4
#SBATCH --mem=32G
#SBATCH --time=01:00:00
echo "=========================================="
echo "Probe 15 Task D: LacdiNAc Permutation Test"
echo "=========================================="
echo "Job ID: $SLURM_JOB_ID"
echo "Node: $HOSTNAME"
echo "GPU: $(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null || echo 'N/A')"
echo "Start: $(date)"
echo ""
# Activate conda
source ~/.bashrc
conda activate glycanml
cd /work/ratul1/supantha/glycan-SD-VS/bert_training_v3/v3.1_cluster_training
python bert_v6_contrastive/scripts/probe_15_seal_milk_ood.py \
--model v6 \
--task d
echo ""
echo "End: $(date)"
echo "=========================================="