Skip to content

Commit 4d6f9c1

Browse files
Muhammad Zaid HameedMuhammad Zaid Hameed
authored andcommitted
correcting parameters description in TRADES trainer base class
Signed-off-by: Muhammad Zaid Hameed <[email protected]>
1 parent 51b78df commit 4d6f9c1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

art/defences/trainer/adversarial_trainer_trades.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@ def __init__(
5757
Create an :class:`.AdversarialTrainerTRADES` instance.
5858
5959
:param classifier: Model to train adversarially.
60-
:param eps: Maximum perturbation that the attacker can introduce.
60+
:param attack: attack to use for data augmentation in adversarial training
61+
:param beta: The scaling factor controlling tradeoff between clean loss and adversarial loss
6162
"""
6263
self._attack = attack
6364
self._beta = beta

0 commit comments

Comments
 (0)