Add configurable wandb log directory path

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

llm_lab/config/train_config.py

@@ -99,6 +99,8 @@ class TrainConfig:
     # ── wandb ──
     wandb_project: str = "llm-1b-lab"
     wandb_run_name: Optional[str] = None
+    wandb_dir: Optional[str] = None
+    """Directory for wandb log files. Defaults to wandb's own default (usually ./wandb)."""
     use_wandb: bool = True
 
     # ── Reproducibility ──

llm_lab/training/metrics.py

@@ -44,13 +44,16 @@ class MetricsTracker:
             import wandb
 
             run_id = resume_id or wandb.util.generate_id()
-            self.wandb_run = wandb.init(
+            init_kwargs = dict(
                 project=self.config.wandb_project,
                 name=self.config.wandb_run_name or f"1b-run-{run_id[:6]}",
                 id=run_id,
                 resume="allow",
                 config=self.config.__dict__,
             )
+            if self.config.wandb_dir:
+                init_kwargs["dir"] = self.config.wandb_dir
+            self.wandb_run = wandb.init(**init_kwargs)
             print(f"[wandb] Initialized: {self.wandb_run.url}")
         except ImportError:
             print("[wandb] Not installed. Using console logging only.")

notebooks/03_training.ipynb

@@ -78,7 +78,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": "# --- 모델 설정 ---\nmodel_config = ModelConfig.debug_10m()  # 검증 시 debug, 실제 학습 시 base_1b()\n\n# --- 데이터 설정 ---\ndata_config = DataConfig(\n    max_seq_len=model_config.max_seq_len,\n    batch_size=4,\n)\n\n# --- 학습 설정 ---\ntrain_config = TrainConfig.debug_10m()  # 검증 시 debug, 실제 학습 시 base_1b()\n\nprint(f\"Effective batch size: {train_config.effective_batch_size}\")\nprint(f\"Total steps: {train_config.total_steps:,}\")"
+   "source": "# --- 모델 설정 ---\nmodel_config = ModelConfig.debug_10m()  # 검증 시 debug, 실제 학습 시 base_1b()\n\n# --- 데이터 설정 ---\ndata_config = DataConfig(\n    max_seq_len=model_config.max_seq_len,\n    batch_size=4,\n)\n\n# --- 학습 설정 ---\ntrain_config = TrainConfig.debug_10m()  # 검증 시 debug, 실제 학습 시 base_1b()\ntrain_config.wandb_dir = \"/content/drive/MyDrive/wandb_logs\"\n\nprint(f\"Effective batch size: {train_config.effective_batch_size}\")\nprint(f\"Total steps: {train_config.total_steps:,}\")"
   },
   {
    "cell_type": "markdown",