Small fix to make PGD work with speech recognizer

HSTEHSTEHSTE · Dec 8, 2023 · 919a6df · 919a6df
1 parent ab15eed
commit 919a6df
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/art/attacks/evasion/projected_gradient_descent/projected_gradient_descent_pytorch.py b/art/attacks/evasion/projected_gradient_descent/projected_gradient_descent_pytorch.py
@@ -61,7 +61,7 @@ class ProjectedGradientDescentPyTorch(ProjectedGradientDescentCommon):
 
     def __init__(
         self,
-        estimator: PyTorchClassifier,
+        estimator: BaseEstimator,
         norm: Union[int, float, str] = np.inf,
         eps: Union[int, float, np.ndarray] = 0.3,
         eps_step: Union[int, float, np.ndarray] = 0.1,
@@ -99,10 +99,10 @@ def __init__(
                                ‘runs/exp1’, ‘runs/exp2’, etc. for each new experiment to compare across them.
         :param verbose: Show progress bars.
         """
-        if not estimator.all_framework_preprocessing:
-            raise NotImplementedError(
-                "The framework-specific implementation only supports framework-specific preprocessing."
-            )
+        # if not estimator.all_framework_preprocessing:
+        #     raise NotImplementedError(
+        #         "The framework-specific implementation only supports framework-specific preprocessing."
+        #     )
 
         if summary_writer and num_random_init > 1:
             raise ValueError("TensorBoard is not yet supported for more than 1 random restart (num_random_init>1).")