Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "9e4dd4b86f77653b59dc1fa97fe3f72e8252b359"
Commit 224f22b9 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

delete debug

parent 66029dd8
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/eval.py
View file @ 224f22b9
...@@ -54,7 +54,8 @@ def main(): ...@@ -54,7 +54,8 @@ 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'])
extra_input = config['Architecture']['algorithm'] in ["SRN", "NRTR", "SAR", "SEED"] extra_input = config['Architecture'][
'algorithm'] in ["SRN", "NRTR", "SAR", "SEED"]
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:
...@@ -68,7 +69,6 @@ def main(): ...@@ -68,7 +69,6 @@ def main():
# build metric # build metric
eval_class = build_metric(config['Metric']) eval_class = build_metric(config['Metric'])
logger.info(f"extra_inputs: {extra_input}")
# 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, extra_input) eval_class, model_type, extra_input)
......
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