5 lines
308 B
Bash
5 lines
308 B
Bash
# This script launches SSD300 training in FP32 on 4 GPUs using 128 batch size (32 per GPU)
|
|
# Usage ./SSD300_FP32_4GPU.sh <path to this repository> <path to dataset> <additional flags>
|
|
|
|
python -m torch.distributed.launch --nproc_per_node=4 $1/main.py --backbone resnet50 --warmup 300 --bs 32 --data $2 ${@:3}
|