From 364c693d267529770520982b24652fa45cb2f7d1 Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Tue, 13 Aug 2024 21:44:21 +0800 Subject: [PATCH] fix #1731 (#1732) Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- sample-apps/endoscopy/lib/trainers/deepedit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sample-apps/endoscopy/lib/trainers/deepedit.py b/sample-apps/endoscopy/lib/trainers/deepedit.py index b8f018b29..ebfb954b6 100644 --- a/sample-apps/endoscopy/lib/trainers/deepedit.py +++ b/sample-apps/endoscopy/lib/trainers/deepedit.py @@ -16,7 +16,7 @@ import torch from monai.apps.deepgrow.transforms import AddRandomGuidanced, FindDiscrepancyRegionsd from monai.handlers import MeanDice, from_engine -from monai.handlers.ignite_metric import IgniteMetric +from monai.handlers.ignite_metric import IgniteMetricHandler from monai.inferers import SimpleInferer from monai.losses import DiceLoss from monai.metrics import MeanIoU @@ -149,7 +149,7 @@ def val_iteration_update(self, context: Context): ) -class MeanIoUMetric(IgniteMetric): +class MeanIoUMetric(IgniteMetricHandler): def __init__( self, include_background: bool = True,