Add debug line for wiki

pomonam · Jul 8, 2024 · 1c42025 · 1c42025
1 parent 1dc7d85
commit 1c42025
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kronfluence/score/dot_product.py b/kronfluence/score/dot_product.py
@@ -82,6 +82,8 @@ def compute_dot_products_with_loader(
         for batch in train_loader:
             batch = send_to_device(tensor=batch, device=state.device)
 
+            print(tracked_module_names)
+
             with no_sync(model=model, state=state):
                 model.zero_grad(set_to_none=True)
                 with autocast(device_type=state.device.type, enabled=enable_amp, dtype=score_args.amp_dtype):