diff --git a/pytorch_influence_functions/calc_influence_function.py b/pytorch_influence_functions/calc_influence_function.py index 85f0f7a..9450db5 100644 --- a/pytorch_influence_functions/calc_influence_function.py +++ b/pytorch_influence_functions/calc_influence_function.py @@ -467,7 +467,8 @@ def calc_img_wise(config, model, train_loader, test_loader): start_time = time.time() influence, harmful, helpful, _ = calc_influence_single( model, train_loader, test_loader, test_id_num=i, gpu=config['gpu'], - recursion_depth=config['recursion_depth'], r=config['r_averaging']) + recursion_depth=config['recursion_depth'], r=config['r_averaging'], + damp=config["damp"], scale=config["scale"]) end_time = time.time() ###########