Skip to content

Commit b42df9d

Browse files
committed
Revert wandb check
1 parent d00d1b6 commit b42df9d

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

notebooks/Colab_WNet3D_training.ipynb

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -396,6 +396,13 @@
396396
" eval_label_folder,\n",
397397
" ) if do_validation else None\n",
398398
"\n",
399+
"try:\n",
400+
" import wandb\n",
401+
" WANDB_INSTALLED = True\n",
402+
"except ImportError:\n",
403+
" WANDB_INSTALLED = False\n",
404+
"\n",
405+
"\n",
399406
"train_config = WNetTrainingWorkerConfig(\n",
400407
" device=\"cuda:0\",\n",
401408
" max_epochs=number_of_epochs,\n",

0 commit comments

Comments
 (0)