Skip to content

Commit 438cb5c

Browse files
committed
support for nondeterministic input shape to a model by gradient matching attack.
Signed-off-by: Taesung Lee <[email protected]>
1 parent 7f1fce6 commit 438cb5c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

art/attacks/poisoning/gradient_matching_attack.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,8 +151,8 @@ def __initialize_poison_tensorflow(
151151
152152
:param x_trigger: A list of samples to use as triggers.
153153
:param y_trigger: A list of target classes to classify the triggers into.
154-
:param x_train: A list of training data to poison a portion of.
155-
:param y_train: A list of labels for x_train.
154+
:param x_poison: A list of training data to poison a portion of.
155+
:param y_poison: A list of true labels for x_poison.
156156
"""
157157
# pylint: disable=no-name-in-module
158158
from tensorflow.keras import backend as K

0 commit comments

Comments
 (0)