We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8299d9d commit b022b39Copy full SHA for b022b39
sparsebit/quantization/tools/calibration.py
@@ -82,13 +82,8 @@ def feature_layerwise_calibration(self, device):
82
for inp_node in node.all_input_nodes:
83
inp_tensors = self.builder.storage.get_output(inp_node.target)
84
for inp_tensor in inp_tensors:
85
-<<<<<<< HEAD
86
if isinstance(inp_tensor, torch.Tensor):
87
module.input_quantizer.update_observer(inp_tensor)
88
-=======
89
- module.input_quantizer.update_observer(inp_tensor)
90
- break #only take first input of operator into consideration
91
->>>>>>> ac38fea... WIP
92
module.input_quantizer.calc_qparams()
93
module.input_quantizer.observer.data_cache.reset()
94
0 commit comments