DeepLearningExamples/PyTorch/Detection/SSD/examples/SSD300_A100_FP16_1GPU.sh
2020-07-07 23:41:00 +02:00

5 lines
258 B
Bash

# This script launches SSD300 training in FP16 on 1 GPUs using 256 batch size
# Usage bash SSD300_FP16_1GPU.sh <path to this repository> <path to dataset> <additional flags>
python $1/main.py --backbone resnet50 --warmup 300 --bs 256 --amp --data $2 ${@:3}