Skip to content

Commit 184df04

Browse files
SangdaeNameunwoosh
authored andcommitted
20220511 metric change 2
1 parent 3bac4de commit 184df04

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

external/anomaly/ote_anomalib/callbacks/progress.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def on_validation_epoch_end(self, trainer, pl_module):
117117
super().on_validation_epoch_end(trainer, pl_module)
118118
score = None
119119
metric = getattr(self.update_progress_callback, 'metric', None)
120-
if metric in list(trainer.logged_metrics.keys()):
120+
if metric in trainer.logged_metrics:
121121
score = float(trainer.logged_metrics[metric])
122122
progress = int(self._get_progress('predict'))
123123
self.update_progress_callback(progress=progress, score=score)

0 commit comments

Comments
 (0)