mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

directly set pytorch metric when it's empty for regression_test.py (#1704)

Browse Source
This commit is contained in:
AllentDan 2023-02-03 11:25:11 +08:00 committed by GitHub
parent 23eed5c265
commit 7d085bee0e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/regression_test.py
View File

@ -316,7 +316,7 @@ def get_pytorch_result(model_name: str, meta_info: dict, checkpoint_path: Path,
pytorch_metric.update(_metrics[ds])
if ds == dataset:
pytorch_metric.update(_metrics)
else:
if pytorch_metric == dict():
pytorch_metric.update(_metrics)
task_name = metafile_metric['Task']
if task_name not in using_task: