From 86afae26835a0ad5778a7d00b02f0e0dbff9f6c3 Mon Sep 17 00:00:00 2001
From: mmcwilliams <hi@mmcwilliams.com>
Date: Wed, 29 Nov 2023 12:57:57 -0500
Subject: [PATCH] Log when not using sm model

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

diff --git a/scripts/train.sh b/scripts/train.sh
index cf0b068..80e7625 100644
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -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