fix warpctc accuracy

4 years ago · bf6f421aa6
parent aa605e23d5
commit bf6f421aa6
1 changed files with 0 additions and 3 deletions
--- 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)