m

Author: pomonam

kronfluence/module/tracker/base.py

@@ -68,6 +68,8 @@ def _preprocess_gradient(self, output_gradient: torch.Tensor, target_dtype: torc
                 output_gradient.mul_(self.module.gradient_scale)
             else:
                 output_gradient = output_gradient * self.module.gradient_scale
+        print("After")
+        print(output_gradient.dtype)
         return output_gradient
 
     def register_hooks(self) -> None: