Merge pull request #116 from bit-bots/fix_tvm_args

Fix tvm args

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "YOEO"
-version = "1.4.2"
+version = "1.4.3"
 description = "A hybrid CNN for object detection and semantic segmentation"
 authors = ["Florian Vahl <[email protected]>", "Jan Gutsche <[email protected]>"]
 

yoeo/scripts/convertONNXModelToTVM.py

@@ -26,7 +26,6 @@ def make_parser():
         type=int,
         help="Input image size")
     parser.add_argument(
-        "-t",
         "--trials",
         default=20000,
         type=int,
@@ -56,7 +55,6 @@ def make_parser():
         type=str,
         help="Path to the tuning records that are created for this optimization")
     parser.add_argument(
-        "-t",
         "--target",
         default="vulkan -from_device=0",
         type=str,
@@ -89,7 +87,7 @@ def run():
     # Build an TVM Compiler model
     tvmc_model = TVMCModel(mod, params)
 
-    
+
     # Tune the model (depending on the hardware and parameters this takes days)
     if not args.no_tuning:
         tvmc.tune(