Update _apply_preprocessing_gradient in loss_gradient

Signed-off-by: Beat Buesser <beat.buesser@ie.ibm.com>
Trusted-AI · May 23, 2022 · 1c81523 · 1c81523
1 parent 25d96be
commit 1c81523
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/art/estimators/speech_recognition/pytorch_deep_speech.py b/art/estimators/speech_recognition/pytorch_deep_speech.py
@@ -492,7 +492,7 @@ def loss_gradient(self, x: np.ndarray, y: np.ndarray, **kwargs) -> np.ndarray:
             results_[:] = list(results)
             results = results_
 
-        results = self._apply_preprocessing_gradient(x_in, results)
+        results = self._apply_preprocessing_gradient(x, results)
 
         if x.dtype != object:
             results = np.array([i for i in results], dtype=x.dtype)  # pylint: disable=R1721