#!/bin/bash
#SBATCH --time=04:00:00
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=4
#SBATCH --mem=64G
#SBATCH --gres=gpu:a100:1
#SBATCH --job-name="probe14_retrieval"
#SBATCH --output="logs/probe14_retrieval_%j.out"

export PYTHONUNBUFFERED=1

echo "============================================"
echo "Probe 14: Embedding-Based Retrieval"
echo "============================================"
echo "Started: $(date)"
echo "Node: $(hostname)"

source /work/ratul1/supantha/miniconda3/etc/profile.d/conda.sh
conda activate glycanml

cd /work/ratul1/supantha/glycan-SD-VS/bert_training_v3/v3.1_cluster_training

python3 -u bert_v6_contrastive/scripts/probe_14_retrieval.py --model v5

echo ""
echo "Completed: $(date)"