supanthadey1's picture
Add BERTose and AFFINose training code release
1d6f391 verified
Raw
History Blame Contribute Delete
514 Bytes
#!/bin/bash
#SBATCH --job-name=tsne_fig
#SBATCH --partition=nova
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --gres=gpu:1
#SBATCH --mem=32G
#SBATCH --time=00:30:00
#SBATCH --output=bert_v6_contrastive/scripts/tsne_figure_%j.out
#SBATCH --error=bert_v6_contrastive/scripts/tsne_figure_%j.err
echo "Start $(date)"
source ~/.bashrc
conda activate glycanml
python bert_v6_contrastive/scripts/generate_tsne_figure.py --model v6
echo "Done $(date)"
ls -lh bert_v6_contrastive/analysis/probe_results_v6/tsne_figures/