From 6e5174fa59dbe7f09a40ffdf490b56523e6d7e73 Mon Sep 17 00:00:00 2001 From: Thong Nguyen Date: Mon, 14 Dec 2020 15:39:26 +0100 Subject: [PATCH] pass "damp" and "scale" to cacl_influence_single --- pytorch_influence_functions/calc_influence_function.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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() ###########