Don't rely on pwd for setting up .cache (ai-safety-foundation#205)

Fix ai-safety-foundation#204
HuFY-dev · Mar 20, 2024 · 8dc963a · 8dc963a
1 parent f005cc1
commit 8dc963a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparse_autoencoder/train/sweep.py b/sparse_autoencoder/train/sweep.py
@@ -197,7 +197,7 @@ def run_training_pipeline(
         source_data: The source data.
         run_name: The name of the run.
     """
-    checkpoint_path = Path("../../.checkpoints")
+    checkpoint_path = Path(__file__).parent.parent.parent / ".checkpoints"
     checkpoint_path.mkdir(exist_ok=True)
 
     random_seed = hyperparameters["random_seed"]