diff --git a/topi/python/topi/cuda/ssd/multibox.py b/topi/python/topi/cuda/ssd/multibox.py index cb92e5ff44c72..599c729d95cae 100644 --- a/topi/python/topi/cuda/ssd/multibox.py +++ b/topi/python/topi/cuda/ssd/multibox.py @@ -438,7 +438,7 @@ def multibox_detection_gpu(cls_prob, loc_pred, anchor, clip=True, threshold=0.01 """ inter_out = multibox_transform_loc(cls_prob, loc_pred, anchor, clip, threshold, variances) - out = non_max_suppression(inter_out[0], inter_out[1], max_output_size = -1, - iou_threshold = nms_threshold, force_suppress = force_suppress, - top_k = nms_topk, return_indices=False) + out = non_max_suppression(inter_out[0], inter_out[1], max_output_size=-1, + iou_threshold=nms_threshold, force_suppress=force_suppress, + top_k=nms_topk, return_indices=False) return out diff --git a/topi/python/topi/vision/ssd/multibox.py b/topi/python/topi/vision/ssd/multibox.py index 81ce808432cba..f2211cb061b65 100644 --- a/topi/python/topi/vision/ssd/multibox.py +++ b/topi/python/topi/vision/ssd/multibox.py @@ -292,7 +292,7 @@ def multibox_detection(cls_prob, loc_pred, anchor, clip=True, threshold=0.01, nm """ inter_out = multibox_transform_loc(cls_prob, loc_pred, anchor, clip, threshold, variances) - out = non_max_suppression(inter_out[0], inter_out[1], max_output_size = -1, - iou_threshold = nms_threshold, force_suppress = force_suppress, - top_k = nms_topk, return_indices=False) + out = non_max_suppression(inter_out[0], inter_out[1], max_output_size=-1, + iou_threshold=nms_threshold, force_suppress=force_suppress, + top_k=nms_topk, return_indices=False) return out