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