[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
zhiqwang · Dec 24, 2021 · 0c4e0ea · 0c4e0ea
1 parent a6ecb2d
commit 0c4e0ea
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/yolort/runtime/yolo_graphsurgeon.py b/yolort/runtime/yolo_graphsurgeon.py
@@ -141,15 +141,15 @@ def register_nms(
         op = "BatchedNMS_TRT"
         attrs = {
             "plugin_version": "1",
-            'shareLocation': True,
-            'backgroundLabelId': -1,  # no background class
-            'numClasses': self.num_classes,
-            'topK': 1024,
-            'keepTopK': detections_per_img,
-            'scoreThreshold': score_thresh,
-            'iouThreshold': nms_thresh,
-            'isNormalized': normalized,
-            'clipBoxes': False,
+            "shareLocation": True,
+            "backgroundLabelId": -1,  # no background class
+            "numClasses": self.num_classes,
+            "topK": 1024,
+            "keepTopK": detections_per_img,
+            "scoreThreshold": score_thresh,
+            "iouThreshold": nms_thresh,
+            "isNormalized": normalized,
+            "clipBoxes": False,
         }
 
         # NMS Outputs