Add debug statement

pomonam · Jul 10, 2024 · 7ea5ca7 · 7ea5ca7
1 parent 3310f19
commit 7ea5ca7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/kronfluence/module/tracker/base.py b/kronfluence/module/tracker/base.py
@@ -61,6 +61,7 @@ def _preprocess_gradient(self, output_gradient: torch.Tensor, target_dtype: torc
                 The preprocessed gradient.
         """
         original_dtype = output_gradient.dtype
+        print(output_gradient.dtype)
         output_gradient = output_gradient.to(dtype=target_dtype)
         if self.module.gradient_scale != 1.0:
             if original_dtype != target_dtype: