Ready to run train jobs. TODO: export and upload model
This commit is contained in:
parent
559e0777cf
commit
849ae58f9b
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue