added reports and params
This commit is contained in:
@@ -211,7 +211,7 @@ def run_unlearning_and_strategy_eval(env_dict, forget_class_idx, strategy, evalu
|
||||
# entry
|
||||
if __name__ == "__main__":
|
||||
|
||||
outer_loop = 1
|
||||
outer_loop = 10
|
||||
inner_loop = CLASS_SIZE
|
||||
|
||||
for k in range(outer_loop):
|
||||
@@ -221,13 +221,13 @@ if __name__ == "__main__":
|
||||
runtime_environment = prepare_data_and_model_environment()
|
||||
|
||||
# Baseline Evaluation
|
||||
finetuning = False
|
||||
finetuning = True
|
||||
# switch finetuning for tests on strategies only
|
||||
run_finetuning_or_baseline_eval(runtime_environment, run_training = finetuning)
|
||||
# scale 16400.0 for ResNet
|
||||
scale = 14500
|
||||
scale = 20200
|
||||
# batch 8 for resNet,
|
||||
unlearning_batches = 32
|
||||
unlearning_batches = 16
|
||||
# regularis
|
||||
# strategies
|
||||
certified_unlearning = CertifiedUnlearning(
|
||||
|
||||
Reference in New Issue
Block a user