Skip to content

Commit

Permalink
Remove non necessary function (open-mmlab#4)
Browse files Browse the repository at this point in the history
  • Loading branch information
kshpv authored Nov 18, 2020
1 parent 2be58f1 commit 667b7a8
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions mmdet/integration/nncf/compression.py
Original file line number Diff line number Diff line change
Expand Up @@ -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():
"""
Expand Down Expand Up @@ -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 (
Expand Down

0 comments on commit 667b7a8

Please sign in to comment.