| # | |
| #SBATCH --job-name=esm.msa | |
| # Default in slurm | |
| #SBATCH [email protected] | |
| #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 | |