minor

pomonam · Jul 8, 2024 · 3f83568 · 3f83568
1 parent 85f99a5
commit 3f83568
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)