Commit 6f728426 authored by Ferreira Da Silva, Rafael's avatar Ferreira Da Silva, Rafael
Browse files

Update job.sb

parent 26e13140
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -2,7 +2,7 @@
#SBATCH -J tft
#SBATCH -t 1:00:00
#SBATCH -A stf053
#SBATCH -N 18
#SBATCH -N 90
#SBATCH -C nvme
#SBATCH --exclusive
##SBATCH --ntasks-per-node=8
@@ -18,7 +18,7 @@ export TRANSFORMERS_CACHE=$(pwd)/.cache

CMD="python -u train.py --config ./topaz_config.yaml"

REPLICAS=2
REPLICAS=10
NODES_PER_EXEC=$((SLURM_NNODES / REPLICAS))

for i in {1..$REPLICAS}