apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

train.py resume argument as store_true

Browse Source 
Default is false.

If want to resume, call train.py --resume
This commit is contained in:
Guillermo García
2018-12-05 14:34:53 +01:00
parent c807c16b79
commit b1fb6fa33d
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/gcp.sh
+2 -2
View File
@@ -4,7 +4,7 @@
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py
# Resume
python3 train.py --resume 1
python3 train.py --resume
# Detect
gsutil cp gs://ultralytics/yolov3.pt yolov3/weights
@@ -24,7 +24,7 @@ python3 test.py --img_size 416 --weights weights/backup5.pt --nms_thres 0.45
# Download and Resume
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3
wget https://storage.googleapis.com/ultralytics/yolov3.pt -O weights/latest.pt
python3 train.py --img_size 416 --batch_size 16 --epochs 1 --resume 1
python3 train.py --img_size 416 --batch_size 16 --epochs 1 --resume
python3 test.py --img_size 416 --weights weights/latest.pt --conf_thres 0.5
# Copy latest.pt to bucket