diff --git a/predtuner/approxes/default_approx_params.json b/predtuner/approxes/default_approx_params.json index b0046a7a96d557987757820869adb251ff9258cd..8e55440dc7caa16e17de9a2941e912af9abbd671 100644 --- a/predtuner/approxes/default_approx_params.json +++ b/predtuner/approxes/default_approx_params.json @@ -5,150 +5,6 @@ "class": "FP16Approx", "name": "12", "exp_speedup": 1.5 -}, { - "direction_is_row": false, - "stride": 2, - "offset": 0, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "121", - "exp_speedup": 2.0 -}, { - "direction_is_row": false, - "stride": 2, - "offset": 1, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "122", - "exp_speedup": 2.0 -}, { - "direction_is_row": true, - "stride": 2, - "offset": 0, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "123", - "exp_speedup": 2.0 -}, { - "direction_is_row": true, - "stride": 2, - "offset": 1, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "124", - "exp_speedup": 2.0 -}, { - "direction_is_row": false, - "stride": 3, - "offset": 0, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "125", - "exp_speedup": 1.5 -}, { - "direction_is_row": false, - "stride": 3, - "offset": 1, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "126", - "exp_speedup": 1.5 -}, { - "direction_is_row": false, - "stride": 3, - "offset": 2, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "127", - "exp_speedup": 1.5 -}, { - "direction_is_row": true, - "stride": 3, - "offset": 0, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "128", - "exp_speedup": 1.5 -}, { - "direction_is_row": true, - "stride": 3, - "offset": 1, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "129", - "exp_speedup": 1.5 -}, { - "direction_is_row": true, - "stride": 3, - "offset": 2, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "130", - "exp_speedup": 1.5 -}, { - "direction_is_row": false, - "stride": 4, - "offset": 0, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "131", - "exp_speedup": 1.33 -}, { - "direction_is_row": false, - "stride": 4, - "offset": 1, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "132", - "exp_speedup": 1.33 -}, { - "direction_is_row": false, - "stride": 4, - "offset": 2, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "133", - "exp_speedup": 1.33 -}, { - "direction_is_row": false, - "stride": 4, - "offset": 3, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "134", - "exp_speedup": 1.33 -}, { - "direction_is_row": true, - "stride": 4, - "offset": 0, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "135", - "exp_speedup": 1.33 -}, { - "direction_is_row": true, - "stride": 4, - "offset": 1, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "136", - "exp_speedup": 1.33 -}, { - "direction_is_row": true, - "stride": 4, - "offset": 2, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "137", - "exp_speedup": 1.33 -}, { - "direction_is_row": true, - "stride": 4, - "offset": 3, - "use_fp16": false, - "class": "PerforateConv2dStride", - "name": "138", - "exp_speedup": 1.33 }, { "direction_is_row": false, "stride": 2, @@ -293,70 +149,6 @@ "class": "PerforateConv2dStride", "name": "168", "exp_speedup": 2.0 -}, { - "skip_every": 2, - "skip_offset": 0, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "231", - "exp_speedup": 2.0 -}, { - "skip_every": 2, - "skip_offset": 1, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "232", - "exp_speedup": 2.0 -}, { - "skip_every": 3, - "skip_offset": 0, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "233", - "exp_speedup": 1.5 -}, { - "skip_every": 3, - "skip_offset": 1, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "234", - "exp_speedup": 1.5 -}, { - "skip_every": 3, - "skip_offset": 2, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "235", - "exp_speedup": 1.5 -}, { - "skip_every": 4, - "skip_offset": 0, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "236", - "exp_speedup": 1.33 -}, { - "skip_every": 4, - "skip_offset": 1, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "237", - "exp_speedup": 1.33 -}, { - "skip_every": 4, - "skip_offset": 2, - "interp_rate": 1.0, - "use_fp16": false, - "class": "Conv2dSampling", - "name": "238", - "exp_speedup": 1.33 }, { "skip_every": 4, "skip_offset": 3, diff --git a/test/test_torchapp.py b/test/test_torchapp.py index 98b941be324cebbc24b5b6b376744045c694561b..12c9121c9c4e8be01fb3c89b862780b6ae601c1f 100644 --- a/test/test_torchapp.py +++ b/test/test_torchapp.py @@ -35,7 +35,7 @@ class TestTorchAppTuning(TorchAppSetUp): self.assertEqual(len(n_knobs), 34) for op_name, op in self.app.midx.name_to_module.items(): if isinstance(op, Conv2d): - nknob = 56 + nknob = 30 elif isinstance(op, Linear): nknob = 2 else: