resolved
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user