Skip to content

Commit 10fe09a

Browse files
committed
Fix dependencies
Signed-off-by: Beat Buesser <[email protected]>
1 parent 4d81448 commit 10fe09a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

art/attacks/evasion/steal_now_attack_later/steal_now_attack_later.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,15 +101,15 @@ def _generate_tile_kernel(patch: list, mask: list, tile_size: int) -> Tuple["tor
101101
repeat = 2
102102
p1_list = []
103103
for _ in range(repeat):
104-
p_1 = torchvision.transforms.Resize((tile_size // 2, new_len))(t_patch)
104+
p_1 = torchvision.transforms.Resize((max(tile_size // 2, 1), new_len))(t_patch)
105105
if torch.rand([]) < 0.6:
106106
p1_list.append(FlipOp(p_1))
107107
else:
108108
p1_list.append(torch.zeros_like(p_1))
109109
p_1 = torch.cat(p1_list, dim=-2)
110110
p_list.append(p_1)
111111

112-
p_2 = torchvision.transforms.RandomCrop((tile_size, tile_size % new_len))(p_1)
112+
p_2 = torchvision.transforms.RandomCrop((tile_size, max(tile_size % new_len, 1)))(p_1)
113113
p_list.append(FlipOp(p_2))
114114

115115
n_patch = torch.cat(p_list, dim=-1)

0 commit comments

Comments
 (0)