|
@@ -76,8 +76,7 @@ def load_model(model_dir):
|
|
|
ratios=model.pruning_ratios,
|
|
ratios=model.pruning_ratios,
|
|
|
axis=paddleslim.dygraph.prune.filter_pruner.FILTER_DIM)
|
|
axis=paddleslim.dygraph.prune.filter_pruner.FILTER_DIM)
|
|
|
|
|
|
|
|
- if status == 'Quantized' or osp.exists(
|
|
|
|
|
- osp.join(model_dir, "quant.yml")):
|
|
|
|
|
|
|
+ if status == 'Quantized':
|
|
|
with open(osp.join(model_dir, "quant.yml")) as f:
|
|
with open(osp.join(model_dir, "quant.yml")) as f:
|
|
|
quant_info = yaml.load(f.read(), Loader=yaml.Loader)
|
|
quant_info = yaml.load(f.read(), Loader=yaml.Loader)
|
|
|
quant_config = quant_info['quant_config']
|
|
quant_config = quant_info['quant_config']
|