Revert wandb check

AdaptiveMotorControlLab · Dec 21, 2024 · b42df9d · b42df9d
1 parent d00d1b6
commit b42df9d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/notebooks/Colab_WNet3D_training.ipynb b/notebooks/Colab_WNet3D_training.ipynb
@@ -396,6 +396,13 @@
     "        eval_label_folder,\n",
     "    ) if do_validation else None\n",
     "\n",
+    "try:\n",
+    "  import wandb\n",
+    "  WANDB_INSTALLED = True\n",
+    "except ImportError:\n",
+    "  WANDB_INSTALLED = False\n",
+    "\n",
+    "\n",
     "train_config = WNetTrainingWorkerConfig(\n",
     "    device=\"cuda:0\",\n",
     "    max_epochs=number_of_epochs,\n",