#!/bin/bash
#SBATCH -J infer_chromnitron
#SBATCH -c2
#SBATCH -p a100_short,a100_long,gpu4_short,gpu4_medium,gpu4_long,gpu8_short,gpu8_medium,gpu8_long
#SBATCH --gres=gpu:1
#SBATCH --mem=20gb  # Job memory request
#SBATCH --time=12:00:00  # Time limit hrs:min:sec
#SBATCH --output=../logs/infer_batched/%x_%j.out
#SBATCH --error=../logs/infer_batched/%x_%j.err
#SBATCH --exclude=a100-4047,a100-4041,a100-4040,a100-4045,a100-4028,gn-0020,gpu-0001,a100-4034,a100-4018,a100-4005

SHARD_ID=$SLURM_ARRAY_TASK_ID

source ~/.bashrc
conda activate chromnitron

cd /gpfs/home/yb2612/yb2612_tsirigos/Chromnitron/git/chromnitron-infer/chromnitron
python -u inference_chunk.py $CONFIG_PATH $SHARD_ID