Ready to run train jobs. TODO: export and upload model

This commit is contained in:
Matt McWilliams 2023-06-29 09:28:10 -04:00
parent 559e0777cf
commit 849ae58f9b
1 changed files with 2 additions and 2 deletions

View File

@ -37,7 +37,7 @@ if [ ! -f "${DEST}" ]; then
wget -q -O "${DEST}" "${DOWNLOAD}"
fi
unzip "${DEST}" -d "${UNZIPPED}"
unzip -o "${DEST}" -d "${UNZIPPED}"
if [ ! -f "${UNZIPPED}/data.yaml" ]; then
fail "${ID}" "Invalid dataset"
@ -56,7 +56,7 @@ WEIGHTS="yolov5s.pt"
if [[ "${MODEL}" == "yolov5m_onnx" ]]; then
WEIGHTS="yolov5m.pt"
else if [[ "${MODEL}" == "yolov5l_onnx" ]]; then
elif [[ "${MODEL}" == "yolov5l_onnx" ]]; then
WEIGHTS="yolov5l.pt"
fi