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 (