diff --git a/pix2pixHD_attack/test.py b/pix2pixHD_attack/test.py index 383ff10..7a0ddb3 100755 --- a/pix2pixHD_attack/test.py +++ b/pix2pixHD_attack/test.py @@ -57,6 +57,7 @@ for i, data in enumerate(dataset): exit(0) minibatch = 1 + # Transfer # if i == 0: # adv_image, perturb = model.attack(data['label'], data['inst'], data['image']) if opt.engine: