diff --git a/src/aijack/attack/labelleakage/normattack.py b/src/aijack/attack/labelleakage/normattack.py index b69d232a..7b63b9fc 100644 --- a/src/aijack/attack/labelleakage/normattack.py +++ b/src/aijack/attack/labelleakage/normattack.py @@ -42,7 +42,7 @@ def attack(self, dataloader): """ epoch_labels = [] epoch_g_norm = [] - for i, data in enumerate(dataloader, 0): + for data in dataloader: inputs, labels = data inputs = inputs.to(self.device) labels = labels.to(self.device)