improve messaging

Author: yztxwd

tpcav/tpcav_model.py

@@ -92,6 +92,8 @@ def fit_pca(
         num_pc: Optional[Union[int, str]] = None,
     ) -> Dict[str, torch.Tensor]:
         """Sample activations, compute PCA, and attach buffers to the model."""
+        logger.info("Start building PCA transformation.")
+
         sampled_avs = []
         for concept in concepts:
             avs = self._sample_concept(concept, num_samples=num_samples_per_concept)
@@ -126,6 +128,8 @@ def fit_pca(
         self._set_buffer("orig_shape", torch.tensor(orig_shape).to(self.device))
         self.fitted = True
 
+        logger.info("PCA transformation built.")
+
         return {
             "zscore_mean": mean,
             "zscore_std": std,