update

alibaba · Oct 24, 2023 · f571062 · f571062
1 parent b5ed643
commit f571062
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easycv/apis/export.py b/easycv/apis/export.py
@@ -276,7 +276,7 @@ def _export_yolox(model, cfg, filename):
                 len(img_scale) == 2
             ), 'Export YoloX predictor config contains img_scale must be (int, int) tuple!'
 
-            input = 255 * torch.rand((batch_size, 3) + img_scale)
+            input = 255 * torch.rand((batch_size, 3) + tuple(img_scale))
 
             # assert use_trt_efficientnms only happens when static_opt=True
             if static_opt is not True: