supanthadey1's picture
Add BERTose and AFFINose training code release
1d6f391 verified
Raw
History Blame Contribute Delete
986 Bytes
#!/bin/bash
#SBATCH --job-name=probe15_taskC2
#SBATCH --output=bert_v6_contrastive/cluster_scripts/slurm_logs/probe15c2_%j.out
#SBATCH --error=bert_v6_contrastive/cluster_scripts/slurm_logs/probe15c2_%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 C v2: Abundance-Weighted Lactation"
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 c2
echo ""
echo "End: $(date)"
echo "=================================================="