@@ -160,7 +160,7 @@ class BaseModel:
mode='w') as f:
yaml.dump(pruning_info, f)
- if self.status == 'Quantized' and self.quanter is not None:
+ if self.status == 'Quantized' and self.quantizer is not None:
quant_info = self.get_quant_info()
with open(
osp.join(save_dir, 'quant.yml'), encoding='utf-8',