multi_gpu

This commit is contained in:
Glenn Jocher 2019-03-18 00:59:24 +02:00
parent 7dba5d0171
commit 32f1def48f
1 changed files with 5 additions and 3 deletions

View File

@ -7,7 +7,9 @@ sudo rm -rf cocoapi && git clone https://github.com/cocodataset/cocoapi && cd co
sudo shutdown
# Start
python3 train.py
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3
cp -r weights yolov3
cd yolov3 && python3 train.py --batch-size 26
# Resume
python3 train.py --resume
@ -18,11 +20,11 @@ python3 detect.py
# Clone branch
sudo rm -rf yolov3 && git clone -b multi_gpu --depth 1 https://github.com/ultralytics/yolov3
cd yolov3 && python3 train.py --batch-size 104
cd yolov3 && python3 train.py --batch-size 26
sudo rm -rf yolov3 && git clone -b multigpu --depth 1 https://github.com/alexpolichroniadis/yolov3
cp coco.data yolov3/cfg
cd yolov3 && python3 train.py --batch-size 104
cd yolov3 && python3 train.py --batch-size 26
# Test
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3