fix activations store innefficiency

jbloomAus · Feb 8, 2024 · 07d38a0 · 07d38a0
1 parent 479765b
commit 07d38a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sae_training/activations_store.py b/sae_training/activations_store.py
@@ -272,7 +272,7 @@ def get_data_loader(
         # 3. put other 50 % in a dataloader
         dataloader = iter(
             DataLoader(
-                mixing_buffer[: mixing_buffer.shape[0] // 2 :],
+                mixing_buffer[mixing_buffer.shape[0] // 2 :],
                 batch_size=batch_size,
                 shuffle=True,
             )