Skip to content

Commit

Permalink
[DLMED] enhance metrics in workflow (#3341)
Browse files Browse the repository at this point in the history
Signed-off-by: Nic Ma <nma@nvidia.com>
  • Loading branch information
Nic-Ma authored Nov 15, 2021
1 parent 8a454bd commit 4d83fc0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion monai/engines/workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,7 +216,7 @@ def _register_metrics(self, k_metric: Dict, add_metrics: Optional[Dict] = None):
if not isinstance(k_metric, dict):
raise TypeError(f"key_metric must be None or a dict but is {type(k_metric).__name__}.")
self.state.key_metric_name = list(k_metric.keys())[0]
metrics = k_metric
metrics = dict(k_metric)
if add_metrics is not None and len(add_metrics) > 0:
if not isinstance(add_metrics, dict):
raise TypeError(f"additional metrics must be None or a dict but is {type(add_metrics).__name__}.")
Expand Down

0 comments on commit 4d83fc0

Please sign in to comment.