fix quant export

truongnn1106 · Jul 5, 2021 · fbbaf53 · fbbaf53
1 parent 605bf83
commit fbbaf53
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py
@@ -92,9 +92,11 @@ def main():
  # build dataloader
  valid_dataloader = build_dataloader(config, 'Eval', device, logger)
 
+ use_srn = config['Architecture']['algorithm'] == "SRN"
+ model_type = config['Architecture']['model_type']
  # start eval
  metirc = program.eval(model, valid_dataloader, post_process_class,
- eval_class)
+ eval_class, model_type, use_srn)
  logger.info('metric eval ***************')
  for k, v in metirc.items():
  logger.info('{}:{}'.format(k, v))