cleaned up for submission
This commit is contained in:
@@ -42,7 +42,7 @@ class Retrain(Strategy):
|
||||
print(f"No naive model found for class {self.target_class_index} retraining a new one")
|
||||
|
||||
|
||||
print(f">> Triggering Exact Unlearning Baseline (Retraining {self.arch.name} from pristine state)...")
|
||||
print(f"Retraining {self.arch.name} from pristine state)...")
|
||||
inner_model = getattr(model, "model", model)
|
||||
if hasattr(inner_model, "fc"):
|
||||
total_classes = inner_model.fc.out_features
|
||||
|
||||
Reference in New Issue
Block a user