#!/bin/bash
#
#SBATCH --job-name=esm.msa
# Default in slurm
#SBATCH --mail-user=gz2294@cumc.columbia.edu
#SBATCH --mail-type=ALL
#SBATCH -t 72:0:0 # Request 5 hours run time
# Define threads and memeory
#SBATCH --ntasks=1
#SBATCH --nodes=1
#SBATCH --cpus-per-task=16        # CPU cores/threads
#SBATCH --mem=40gb              # memory per node
#SBATCH --output=out.log
export OMP_NUM_THREADS=$SLURM_CPUS_PER_TASK

dataset=$1
predictor=$2
n_seeds=1
n_train=-1
n_threads=16
kwargs=

/share/descartes/Users/gz2294/miniconda3/envs/protein_fitness_prediction/bin/python src/evaluate.py $dataset $predictor \
	--n_threads=$n_threads --n_seeds=$n_seeds \
	--n_train=$n_train --joint_training $kwargs