This commit is contained in:
2026-07-01 22:01:21 +02:00
5 changed files with 141 additions and 1 deletions

View File

@@ -50,7 +50,6 @@ class WeightFiltration(Strategy):
for epoch in range(self.epochs):
t_loss_r, t_loss_f = 0.0, 0.0
steps = 0
# forget and retain
for (r_inputs, r_labels), (f_inputs, f_labels) in zip(retain_loader, forget_loader):
r_inputs, r_labels = r_inputs.to(device), r_labels.to(device)