Skip to content

GitLab

Commit 560f2f49 authored by tink2123's avatar tink2123
Browse files

fix eval

parent e885b57e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
tools/eval.py
View file @ 560f2f49
...@@ -54,8 +54,7 @@ def main(): ...@@ -54,8 +54,7 @@ def main():
config['Architecture']["Head"]['out_channels'] = char_num config['Architecture']["Head"]['out_channels'] = char_num
model = build_model(config['Architecture']) model = build_model(config['Architecture'])
use_srn = config['Architecture']['algorithm'] == "SRN" extra_input = config['Architecture']['algorithm'] in ["SRN", "SAR"]
use_sar = config['Architecture']['algorithm'] == "SAR"
if "model_type" in config['Architecture'].keys(): if "model_type" in config['Architecture'].keys():
model_type = config['Architecture']['model_type'] model_type = config['Architecture']['model_type']
else: else:
...@@ -72,7 +71,7 @@ def main(): ...@@ -72,7 +71,7 @@ def main():
# start eval # start eval
metric = program.eval(model, valid_dataloader, post_process_class, metric = program.eval(model, valid_dataloader, post_process_class,
eval_class, model_type, use_srn, use_sar) eval_class, model_type, extra_input)
logger.info('metric eval ***************') logger.info('metric eval ***************')
for k, v in metric.items(): for k, v in metric.items():
logger.info('{}:{}'.format(k, v)) logger.info('{}:{}'.format(k, v))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment