Commit 26e13140 authored by Ferreira Da Silva, Rafael's avatar Ferreira Da Silva, Rafael
Browse files

migrating to ROCm 6.0.0

parent 447cf502

job.sb

+2 −2
Original line number Diff line number Diff line 
#!/bin/bash -l

#SBATCH -J tft

#SBATCH -t 2:00:00

#SBATCH -t 1:00:00

#SBATCH -A stf053

#SBATCH -N 18

#SBATCH -C nvme
@@ -21,7 +21,7 @@ CMD="python -u train.py --config ./topaz_config.yaml" 
REPLICAS=2

NODES_PER_EXEC=$((SLURM_NNODES / REPLICAS))



for i in {1..4}

for i in {1..$REPLICAS}

do

    HOME=/tmp time srun --nodes=${NODES_PER_EXEC} \

                   --ntasks=72 \