Update notebook

truefoundry · Dec 22, 2023 · 39bc287 · 39bc287
1 parent 9bd1e36
commit 39bc287
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/finetune.ipynb b/finetune.ipynb
@@ -265,7 +265,7 @@
     "    from urllib.parse import urljoin\n",
     "    from tensorboard import notebook\n",
     "\n",
-    "    tb_logs = os.path.join(output_dir, \"runs\")\n",
+    "    tb_logs = os.path.join(\".\", \"tensorboard_logs\")\n",
     "    os.makedirs(tb_logs, exist_ok=True)\n",
     "    os.environ[\"TENSORBOARD_PROXY_URL\"] = urljoin(os.getenv(\"NB_PREFIX\", \"/\"), \"proxy/%PORT%/\")\n",
     "    notebook.start(f\"--logdir {tb_logs} --reload_interval 30.0 --reload_multifile True\")\n",