Skip to content

Commit d4388dd

Browse files
authored
fix resnet50_ipex tuning (#1295)
1 parent d13d446 commit d4388dd

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

neural_compressor/adaptor/pytorch.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2244,9 +2244,6 @@ def quantize(self, tune_cfg, model, dataloader, q_func=None):
22442244
except:
22452245
q_model = torch.jit.trace(q_model, example_inputs, strict=False)
22462246
q_model = torch.jit.freeze(q_model)
2247-
#to enable fuse ops
2248-
self.model_calibration(q_model, dataloader, iterations=2, conf=None,
2249-
calib_sampling_size = tune_cfg.get('calib_sampling_size', 1))
22502247
assert self.approach != 'quant_aware_training', \
22512248
"Intel PyTorch Extension didn't support quantization aware training mode"
22522249
model_._model = q_model

0 commit comments

Comments
 (0)