From 1a5125b36ef44b6b2c5cc45b6b57d0e6e9d9dce1 Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Tue, 25 Jan 2022 02:24:47 +0800 Subject: [PATCH] Set enable_dynamic as False by default --- yolort/runtime/yolo_graphsurgeon.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/yolort/runtime/yolo_graphsurgeon.py b/yolort/runtime/yolo_graphsurgeon.py index ce25b07e..d151caa4 100644 --- a/yolort/runtime/yolo_graphsurgeon.py +++ b/yolort/runtime/yolo_graphsurgeon.py @@ -48,7 +48,7 @@ class YOLOGraphSurgeon: score_thresh (float): Score threshold used for postprocessing the detections. version (str): upstream version released by the ultralytics/yolov5, Possible values are ["r3.1", "r4.0", "r6.0"]. Default: "r6.0". - enable_dynamic (bool): Whether to specify axes of tensors as dynamic. Default: True. + enable_dynamic (bool): Whether to specify axes of tensors as dynamic. Default: False. device (torch.device): The device to be used for importing ONNX. Default: torch.device("cpu"). """