DeepLearningExamples/PyTorch/SpeechRecognition/Jasper/platform/DGX2_Jasper_AMP_16GPU.sh
2020-07-07 22:58:04 +02:00

4 lines
105 B
Bash

#!/bin/bash
NUM_GPUS=16 AMP=true BATCH_SIZE=64 GRADIENT_ACCUMULATION_STEPS=1 bash scripts/train.sh "$@"