config_singlenode.sh

#!/bin/bash

## DL params
EXTRA_PARAMS=(
               --batch-size      "180"
               --warmup          "2.619685" # 300 iterations * 8 GPUs * 1 nodes * 128 batch size / 117266 non-empty images
             )

## System run parms
DGXNNODES=1
DGXSYSTEM=$(basename $(readlink -f ${BASH_SOURCE[0]}) | sed 's/^config_//' | sed 's/\.sh$//' )
WALLTIME=12:00:00

## System config params
DGXNGPU=4
DGXSOCKETCORES=8
DGXNSOCKET=4
DGXHT=1 	# HT is on is 2, HT off is 1
DGXIBDEVICES=''