diff --git a/pyproject.toml b/pyproject.toml index e39a356..b5f7e02 100644 --- a/pyproject.toml +++ b/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 ", "Jan Gutsche "] diff --git a/yoeo/scripts/convertONNXModelToTVM.py b/yoeo/scripts/convertONNXModelToTVM.py index e39b7de..3dfd70d 100644 --- a/yoeo/scripts/convertONNXModelToTVM.py +++ b/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(