supanthadey1's picture
Add BERTose and AFFINose training code release
1d6f391 verified
Raw
History Blame Contribute Delete
2.16 kB
#!/bin/bash
#SBATCH --job-name=probe10v2
#SBATCH --partition=nova
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=4
#SBATCH --gres=gpu:v100:1
#SBATCH --mem=32G
#SBATCH --time=00:30:00
#SBATCH --output=bert_v6_contrastive/additional_probes/probe10v2_%j.out
#SBATCH --error=bert_v6_contrastive/additional_probes/probe10v2_%j.err
echo "========================================"
echo "Probe 10 v2: Tissue-Specific Glycan Presence"
echo " Data: 243 glycans (best WURCS + recovered)"
echo " Well-powered: 16 tissues (>=30 positives)"
echo "Job ID: $SLURM_JOB_ID"
echo "Node: $HOSTNAME"
echo "Start: $(date)"
echo "========================================"
source activate blade 2>/dev/null || conda activate blade
cd /work/ratul1/supantha/glycan-SD-VS/bert_training_v3/v3.1_cluster_training
SCRIPT=bert_v6_contrastive/additional_probes/probe_10_tissue_presence.py
DATA_V2=bert_v6_contrastive/additional_probes/probe_10_tissue_presence_data_v2.csv
python3 -c "import torch; print(f'PyTorch: {torch.__version__} CUDA: {torch.cuda.is_available()}')"
python3 -c "import torch; print(f'{torch.cuda.get_device_name(0)}, {torch.cuda.get_device_properties(0).total_memory//1024**2} MiB')" 2>/dev/null
echo ""
ls -lh $DATA_V2
wc -l $DATA_V2
echo ""
echo "========== V5-A: v2 data, all 23 tissues =========="
python3 $SCRIPT --model v5 --data-version v2
echo ""
echo "========== V5-A: v2 data, well-powered (16 tissues) =========="
python3 $SCRIPT --model v5 --data-version v2 --well-powered
echo ""
echo "========== V6: v2 data, all 23 tissues =========="
python3 $SCRIPT --model v6 --data-version v2
echo ""
echo "========== V6: v2 data, well-powered (16 tissues) =========="
python3 $SCRIPT --model v6 --data-version v2 --well-powered
echo ""
echo "Completed: $(date)"
echo "Output dirs:"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v5_v2/ 2>/dev/null
echo "---"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v5_v2_wp/ 2>/dev/null
echo "---"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v6_v2/ 2>/dev/null
echo "---"
ls -lh bert_v6_contrastive/additional_probes/probe10_results_v6_v2_wp/ 2>/dev/null