From 849ae58f9b1fd0e4578a06a0a2230e99d7a49376 Mon Sep 17 00:00:00 2001
From: mattmcw <matt@sixteenmillimeter.com>
Date: Thu, 29 Jun 2023 09:28:10 -0400
Subject: [PATCH] Ready to run train jobs. TODO: export and upload model

---
 scripts/train.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/train.sh b/scripts/train.sh
index d53b119..0ee2565 100644
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -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