From 667b7a811c444aa228b3ffa8b3f9bbef4215a0de Mon Sep 17 00:00:00 2001 From: Aleksei Kashapov Date: Wed, 18 Nov 2020 12:57:53 +0300 Subject: [PATCH] Remove non necessary function (#4) --- mmdet/integration/nncf/compression.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/mmdet/integration/nncf/compression.py b/mmdet/integration/nncf/compression.py index 1b1b398c4c5..be87ba691d4 100644 --- a/mmdet/integration/nncf/compression.py +++ b/mmdet/integration/nncf/compression.py @@ -35,10 +35,6 @@ def get_inputs(self, dataloader_output): kwargs = {k: v.data[0] for k, v in dataloader_output.items()} return (), kwargs - # TODO: not tested; need to test - def get_target(self, dataloader_output): - return dataloader_output['gt_bboxes'], dataloader_output['gt_labels'] - def get_nncf_metadata(): """ @@ -160,9 +156,9 @@ def dummy_forward(model): def change_export_func_first_conv(model): - ''' To avoid saturation issue + """ To avoid saturation issue At the moment works only for mobilenet - ''' + """ def run_hacked_export_quantization(self, x): from nncf.quantization.layers import (