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