diff --git a/python/tvm/relay/op/vision/nms.py b/python/tvm/relay/op/vision/nms.py index 46ec1e449e124..677e152a6f191 100644 --- a/python/tvm/relay/op/vision/nms.py +++ b/python/tvm/relay/op/vision/nms.py @@ -85,5 +85,5 @@ def non_max_suppression(data, """ return _make.non_max_suppression(data, valid_count, max_output_size, iou_threshold, force_suppress, top_k, - coord_start, score_index, id_index, + coord_start, score_index, id_index, return_indices, invalid_to_bottom)