From bf6f421aa6fb188d81e2f5eb637e034e5b17d4b1 Mon Sep 17 00:00:00 2001 From: baihuawei Date: Thu, 15 Oct 2020 09:23:03 +0800 Subject: [PATCH] fix warpctc accuracy --- model_zoo/official/cv/warpctc/src/metric.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/model_zoo/official/cv/warpctc/src/metric.py b/model_zoo/official/cv/warpctc/src/metric.py index 8e1ca77065..d535b9a4c4 100755 --- a/model_zoo/official/cv/warpctc/src/metric.py +++ b/model_zoo/official/cv/warpctc/src/metric.py @@ -39,9 +39,6 @@ class WarpCTCAccuracy(nn.Metric): y_pred = self._convert_data(inputs[0]) y = self._convert_data(inputs[1]) - if self.device_target == 'GPU': - y = y[:, :-1] - self._count += 1 pred_lbls = self._get_prediction(y_pred)