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

4 lines
104 B
Bash

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