Skip to content

Commit 3e2e310

Browse files
hotfix: X pattern generation logic
1 parent 914c7a0 commit 3e2e310

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

nebula/addons/attacks/dataset/datapoison.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ def add_x_to_image(self, img: torch.Tensor | Image.Image) -> torch.Tensor:
254254
# Create X pattern
255255
for i in range(0, 10):
256256
for j in range(0, 10):
257-
if i + j <= 9 or i == j:
257+
if i + j == 9 or i == j:
258258
if len(img.shape) == 3: # RGB image
259259
img[..., i, j] = pattern_value
260260
else: # Grayscale image

0 commit comments

Comments
 (0)