Skip to content

Commit 5508657

Browse files
dxoigmnmzweilin
andauthored
Adversary as pl.LightningModule (#103)
* Turn Adversary into LightningModule so we can fit perturbations. * Make adversary-specific callbacks generic PL callbacks * Improve progress bar for adversaries (#153) --------- Co-authored-by: Weilin Xu <[email protected]>
1 parent 3d5d5c7 commit 5508657

24 files changed

+400
-458
lines changed

mart/attack/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
11
from .adversary import *
22
from .adversary_in_art import *
33
from .adversary_wrapper import *
4-
from .callbacks import Callback
54
from .composer import *
65
from .enforcer import *
76
from .gain import *
87
from .gradient_modifier import *
98
from .initializer import *
10-
from .objective import Objective
9+
from .objective import *
1110
from .perturber import *
1211
from .projector import *

0 commit comments

Comments
 (0)