From 3f83568820c3b060ba05e3ec1bec3b13d49b9964 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Mon, 8 Jul 2024 16:59:11 -0400 Subject: [PATCH] minor --- kronfluence/score/dot_product.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kronfluence/score/dot_product.py b/kronfluence/score/dot_product.py index 0569e41..961b485 100644 --- a/kronfluence/score/dot_product.py +++ b/kronfluence/score/dot_product.py @@ -82,7 +82,7 @@ def compute_dot_products_with_loader( for batch in train_loader: batch = send_to_device(tensor=batch, device=state.device) - print(cached_module_lst[0]) + print(cached_module_lst[0].storage) with no_sync(model=model, state=state): model.zero_grad(set_to_none=True)