rename notebook, add early stopping for clay finetuned variant

Clay-foundation · Jan 13, 2024 · f37284b · f37284b
1 parent 7292436
commit f37284b
Showing 1 changed file with 4 additions and 54 deletions.
diff --git a/notebooks/flood_benchmark_segmentation.ipynb b/notebooks/flood_benchmark_segmentation.ipynb
@@ -3119,7 +3119,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 70,
+      "execution_count": null,
       "id": "8jqeBd6LOLkr",
       "metadata": {
         "colab": {
@@ -3128,32 +3128,7 @@
         "id": "8jqeBd6LOLkr",
         "outputId": "36bcccb7-ef5c-4c78-e9b8-c02eaf85ea0a"
       },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "IoU: 0.3379289507865906\n",
-            "IoU: 0.5677745938301086\n",
-            "IoU: 0.38128602504730225\n",
-            "IoU: 0.4337167739868164\n",
-            "IoU: 0.07514558732509613\n",
-            "IoU: 0.8329333066940308\n",
-            "IoU: 0.16996563971042633\n",
-            "IoU: 0.2993166744709015\n",
-            "IoU: 0.8047876954078674\n",
-            "IoU: 0.36587828397750854\n",
-            "IoU: 0.9110873341560364\n",
-            "IoU: 0.015298277139663696\n",
-            "IoU: 0.8980720043182373\n",
-            "IoU: 0.08573030680418015\n",
-            "IoU: 0.4487551748752594\n",
-            "IoU: 0.8745391964912415\n",
-            "IoU: 0.18230441212654114\n",
-            "Average IoU: 0.45203059911727905\n"
-          ]
-        }
-      ],
+      "outputs": [],
       "source": [
         "IoUs_fs = calculate_IoU(det_model, dm.val_dataloader(), clay=False)\n",
         "\n",
@@ -3164,7 +3139,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 71,
+      "execution_count": null,
       "id": "6FPXD6tmKdB2",
       "metadata": {
         "colab": {
@@ -3173,32 +3148,7 @@
         "id": "6FPXD6tmKdB2",
         "outputId": "bcd5d8cd-b66f-44c2-dea4-bf75dfe6b8ee"
       },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "IoU: 0.0901893898844719\n",
-            "IoU: 0.4645256996154785\n",
-            "IoU: 0.3826591372489929\n",
-            "IoU: 0.0009378832764923573\n",
-            "IoU: 0.06925993412733078\n",
-            "IoU: 0.6750580072402954\n",
-            "IoU: 0.32923126220703125\n",
-            "IoU: 0.2249605357646942\n",
-            "IoU: 0.8142861723899841\n",
-            "IoU: 0.3637990951538086\n",
-            "IoU: 0.28369003534317017\n",
-            "IoU: 0.056073836982250214\n",
-            "IoU: 0.778351366519928\n",
-            "IoU: 0.20255421102046967\n",
-            "IoU: 0.2703555226325989\n",
-            "IoU: 0.7836553454399109\n",
-            "IoU: 0.08511418849229813\n",
-            "Average IoU from model finetuned from Clay: 0.3455706834793091\n"
-          ]
-        }
-      ],
+      "outputs": [],
       "source": [
         "IoUs_cl = calculate_IoU(det_model_clay, dm.val_dataloader(), clay=True)\n",
         "\n",