WF net added

This commit is contained in:
2026-06-28 01:48:49 +02:00
parent 0680a920ff
commit 0a7a2e1da5
2 changed files with 129 additions and 3 deletions

View File

@@ -20,7 +20,7 @@ BATCH_SIZE = 16
SAMPLE_SIZE = 30
TRAINING_SAMPLE = 27
RESOLUTION = 224
ARCH = Architecture.RESNET18
ARCH = Architecture.GOOGLENET
# Data preparation and model setup
@@ -230,8 +230,8 @@ if __name__ == "__main__":
strategies = [
certified_unlearning,
weight_filtration,
linear_filtration
#weight_filtration,
#linear_filtration
]
# Unlearning Iteration
for i in range(0, CLASS_SIZE):