will-jl944 преди 4 години
родител
ревизия
598a8d446d
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      dygraph/paddlex/cv/models/base.py

+ 1 - 1
dygraph/paddlex/cv/models/base.py

@@ -160,7 +160,7 @@ class BaseModel:
                     mode='w') as f:
                     mode='w') as f:
                 yaml.dump(pruning_info, 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()
             quant_info = self.get_quant_info()
             with open(
             with open(
                     osp.join(save_dir, 'quant.yml'), encoding='utf-8',
                     osp.join(save_dir, 'quant.yml'), encoding='utf-8',