diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index 100b107a1deb1ce9932c9cefa50659c060f5803e..942f59e000a24faef72844b45e833fc55061d2ac 100755 --- 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))