Skip to content

Commit 06eddee

Browse files
committed
Optimize print information for FGSM & PGD
1 parent 0703d71 commit 06eddee

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

cvm/attacks/fgsm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -31,4 +31,4 @@ def perturb(self, images: torch.Tensor, labels: torch.Tensor = None, targeted: b
3131
return self.unprepare_inputs(images_adv)
3232

3333
def __repr__(self) -> str:
34-
return f'FGSM(eps={self.epsilon:>6.4f}, normalized={self.normalized}, mean={self.mean}, std={self.std})'
34+
return f'FGSM(eps={self.epsilon:>6.4f}({self.epsilon * 255.0:>.1f}/255.0), normalized={self.normalized}, mean={self.mean}, std={self.std})'

cvm/attacks/pgd.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,4 +39,4 @@ def perturb(self, images: torch.Tensor, labels: torch.Tensor = None, targeted: b
3939
return self.unprepare_inputs(images_adv)
4040

4141
def __repr__(self) -> str:
42-
return f'PGD(eps={self.epsilon:>6.4f}, steps={self.steps}, alpha={self.alpha:>6.4f}, normalized={self.normalized}, mean={self.mean}, std={self.std})'
42+
return f'PGD(eps={self.epsilon:>6.4f}({self.epsilon * 255.0:>.1f}/255.0), steps={self.steps}, alpha={self.alpha:>6.4f}({self.alpha * 255.0:>.1f}/255.0), normalized={self.normalized}, mean={self.mean}, std={self.std})'

0 commit comments

Comments
 (0)