Ensure max_epochs>=2 for Auto3Dseg notebooks in CI/CD (#955)

Signed-off-by: Mingxin Zheng
<18563433+mingxin-zheng@users.noreply.github.com>

Fixes #952

Signed-off-by: Mingxin Zheng <18563433+mingxin-zheng@users.noreply.github.com>

Author: mingxin-zheng

auto3dseg/notebooks/auto3dseg_autorunner_ref_api.ipynb

@@ -329,6 +329,9 @@
     "\n",
     "max_epochs = 2000\n",
     "\n",
+    "# safeguard to ensure max_epochs is greater or equal to 2\n",
+    "max_epochs = max(max_epochs, 2)\n",
+    "\n",
     "num_gpus = 1 if \"multigpu\" in data_src_cfg and not data_src_cfg[\"multigpu\"] else torch.cuda.device_count()\n",
     "\n",
     "num_epoch = max_epochs\n",

auto3dseg/notebooks/auto3dseg_hello_world.ipynb

@@ -232,6 +232,9 @@
    "source": [
     "max_epochs = 2\n",
     "\n",
+    "# safeguard to ensure max_epochs is greater or equal to 2\n",
+    "max_epochs = max(max_epochs, 2)\n",
+    "\n",
     "train_param = {\n",
     "    \"CUDA_VISIBLE_DEVICES\": [0],  # use only 1 gpu\n",
     "    \"num_iterations\": 4 * max_epochs,\n",

auto3dseg/notebooks/ensemble_byoc.ipynb

@@ -699,6 +699,9 @@
     "\n",
     "max_epochs = 2\n",
     "\n",
+    "# safeguard to ensure max_epochs is greater or equal to 2\n",
+    "max_epochs = max(max_epochs, 2)\n",
+    "\n",
     "train_param = {\n",
     "    \"CUDA_VISIBLE_DEVICES\": [0],  # use only 1 gpu\n",
     "    \"num_iterations\": 4 * max_epochs,\n",

auto3dseg/notebooks/hpo_nni.ipynb

@@ -252,6 +252,9 @@
     "\n",
     "max_epochs = 2\n",
     "\n",
+    "# safeguard to ensure max_epochs is greater or equal to 2\n",
+    "max_epochs = max(max_epochs, 2)\n",
+    "\n",
     "num_gpus = 1 if \"multigpu\" in input_dict and not input_dict[\"multigpu\"] else torch.cuda.device_count()\n",
     "\n",
     "num_epoch = max_epochs\n",

auto3dseg/notebooks/hpo_optuna.ipynb

@@ -255,6 +255,9 @@
     "\n",
     "max_epochs = 2\n",
     "\n",
+    "# safeguard to ensure max_epochs is greater or equal to 2\n",
+    "max_epochs = max(max_epochs, 2)\n",
+    "\n",
     "num_gpus = 1 if \"multigpu\" in input_dict and not input_dict[\"multigpu\"] else torch.cuda.device_count()\n",
     "\n",
     "num_epoch = max_epochs\n",