update

alibaba · Oct 25, 2023 · 699a850 · 699a850
1 parent d5e80f5
commit 699a850
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/easycv/apis/export.py b/easycv/apis/export.py
@@ -369,16 +369,15 @@ def _export_yolox(model, cfg, filename):
                     json.dump(config, ofile)
 
                 torch.onnx.export(
-                    model,  
-                    input.to(device), 
+                    model,
+                    input.to(device),
                     filename if filename.endswith('onnx') else filename +
-                    '.onnx',  
-                    export_params=
-                    True,  
-                    opset_version=12,  
-                    do_constant_folding=True,  
-                    input_names=['input'],  
-                    output_names=['output'],  
+                    '.onnx',
+                    export_params=True,
+                    opset_version=12,
+                    do_constant_folding=True,
+                    input_names=['input'],
+                    output_names=['output'],
                 )
 
             if export_type == 'jit':