DeepLearningExamples/PyTorch/Detection/SSD/qa/testing_DGX1V_8GPU_fp16_full_run.sh

5 lines
282 B
Bash

#!/bin/bash
python3 -m torch.distributed.launch --nproc_per_node=8 qa/qa_accuracy_main.py --bs 64 --fp16 --warmup 300 --learning-rate 2.6e-3 --seed 1 --benchmark-mode full-accuracy --benchmark-file qa/curve_baselines/SSD300_pytorch_18.08_fp16_full_run_acc_baseline.json --data $1