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