fix path for profiler

giotto-ai · May 12, 2022 · 74b08f8 · 74b08f8
1 parent 1cf77c7
commit 74b08f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 40 deletions.
diff --git a/examples/basic_tutorial_tabular.ipynb b/examples/basic_tutorial_tabular.ipynb
@@ -17,45 +17,6 @@
     " 5. visualise results interactively"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import torch\n",
-    "X = torch.rand(3,10,2)\n",
-    "out = torch.stack([(X[i] + torch.rand(10,2))/(torch.rand(10,2)) for i in range(len(X))])\n",
-    "out.shape"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "torch.mean(out, 0, False).shape"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "torch.tensor([1,2])/torch.tensor([1,3])"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "all(torch.tensor([1,-2])>0)"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,

diff --git a/gdeep/pipeline/pipeline.py b/gdeep/pipeline/pipeline.py
@@ -411,7 +411,8 @@ def _init_profiler(self, profiling, cross_validation, n_epochs, k_folds):
                                 torch.profiler.ProfilerActivity.CUDA],
                             schedule=torch.profiler.schedule(wait=1, warmup=1, active=active, repeat=1),
                             on_trace_ready=torch.profiler.tensorboard_trace_handler(
-                                './runs/' + type(self.model).__name__ + str(datetime.today()),
+                                os.path.join('.','runs', (self.model.__class__.__name__ +
+                                                          str(datetime.today())).replace(":","-")),
                                 worker_name='worker'),
                             record_shapes=True,
                             profile_memory=True