diff --git a/utils/gcp.sh b/utils/gcp.sh
index af9de6ee..ad90d3aa 100755
--- a/utils/gcp.sh
+++ b/utils/gcp.sh
@@ -55,7 +55,7 @@ gsutil cp results.png gs://ultralytics
 sudo shutdown
 
 # Reproduce mAP
-python3 test.py --save-json --img-size 608  --batch-size 16
+python3 test.py --save-json --img-size 608
 python3 test.py --save-json --img-size 416
 python3 test.py --save-json --img-size 320
 sudo shutdown
diff --git a/utils/utils.py b/utils/utils.py
index b0047217..d00b8aa4 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -431,7 +431,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
                     dc = dc[i == 0]
 
             elif nms_style == 'SOFT':  # soft-NMS https://arxiv.org/abs/1704.04503
-                sigma = nms_thres  # soft-nms sigma parameter
+                sigma = 0.5  # soft-nms sigma parameter
                 while len(dc):
                     if len(dc) == 1:
                         det_max.append(dc)