mattmcw
/
yolo_web
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Log when not using sm model

This commit is contained in:
mmcwilliams 2023-11-29 12:57:57 -05:00
parent 255526cc3b
commit 86afae2683
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/train.sh
View File

@ -57,8 +57,10 @@ source env/bin/activate
WEIGHTS="yolov5s.pt"
if [[ "${MODEL}" == "yolov5m_onnx" ]]; then
echo "Using medium weights [${MODEL}]"
WEIGHTS="yolov5m.pt"
elif [[ "${MODEL}" == "yolov5l_onnx" ]]; then
echo "Using large weights [${MODEL}]"
WEIGHTS="yolov5l.pt"
fi