add vscode debug config

ValerianRey · ValerianRey · commit bb38440ea70c · 2025-12-22T14:59:39.000Z
diff --git a/.gitignore b/.gitignore
@@ -3,7 +3,7 @@ data/
 checkpoints/
 wandb/
 uv.lock/
-.vscode/
+.vscode/settings.json
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -0,0 +1,39 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Pretrain Sudoku",
+            "type": "debugpy",
+            "request": "launch",
+            "module": "recursion.pretrain",
+            "python": "${workspaceFolder}/.venv/bin/python",
+            "cwd": "${workspaceFolder}",
+            "args": [
+                "arch=trm",
+                "data_paths=[data/sudoku-extreme-1k-aug-1000]",
+                "evaluators=[]",
+                "epochs=5",
+                "eval_interval=1",
+                "lr=1e-4",
+                "puzzle_emb_lr=1e-4",
+                "weight_decay=1.0",
+                "puzzle_emb_weight_decay=1.0",
+                "arch.mlp_t=True",
+                "arch.pos_encodings=none",
+                "arch.L_layers=2",
+                "arch.H_cycles=1",
+                "arch.L_cycles=1",
+                "+run_name=debug",
+                "ema=True"
+            ],
+            "console": "integratedTerminal",
+            "justMyCode": false,
+            "env": {
+                "DISABLE_COMPILE": "1"
+            }
+        }
+    ]
+}
