clean up

ai-safety-foundation · Feb 6, 2024 · c491e11 · c491e11
1 parent 8b82a79
commit c491e11
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/sparse_autoencoder/train/pipeline.py b/sparse_autoencoder/train/pipeline.py
@@ -177,8 +177,6 @@ def generate_activations(self, store_size: PositiveInt) -> TensorActivationStore
             hook = partial(store_activations_hook, store=store, component_idx=component_idx)
             self.source_model.add_hook(cache_name, hook)
 
-        print("Generate Process Begin")
-
         # Loop through the dataloader until the store reaches the desired size
         with torch.no_grad():
             while len(store) < store_size: