diff --git a/rainfallwrangler/slurm-jsonl2tfrecord.job b/rainfallwrangler/slurm-jsonl2tfrecord.job index 47f0f95..bda994e 100755 --- a/rainfallwrangler/slurm-jsonl2tfrecord.job +++ b/rainfallwrangler/slurm-jsonl2tfrecord.job @@ -2,6 +2,7 @@ #SBATCH -J Json2TfR #SBATCH -N 1 #SBATCH -n 28 +#SBATCH --gres=gpu:1 #SBATCH -o %j.%N.%a.out #SBATCH -e %j.%N.%a.err #SBATCH -p gpu05,gpu