AUSLab-AdversarialML/stuff.py at main · agoston-mc/AUSLab-AdversarialML · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import AttackPack

eps_dat = [0.1, 0.01, 0.001, 0.0001]
eps_mix = [0.1, 0.01]

for mname in AttackPack.methods.keys():

    AttackPack.main(
        method = mname,
        eps = eps_dat,
        device = "cuda",
        weights = "vpnet-hermite2-rr_b4096_lr0.01_e50_p1.0-0.0_n8_h8_r0.0.pt",
        adv_method="d",
        # early_stop=1,
    )
    AttackPack.main(
        method=mname,
        eps=eps_mix,
        device="cuda",
        weights="vpnet-hermite2-rr_b4096_lr0.01_e50_p1.0-0.0_n8_h8_r0.0.pt",
        adv_method="r",
        # early_stop=1,
    )
    AttackPack.main(
        method=mname,
        eps=eps_mix,
        device="cuda",
        weights="vpnet-hermite2-rr_b4096_lr0.01_e50_p1.0-0.0_n8_h8_r0.0.pt",
        adv_method="dr",
        # early_stop=1,
    )