modif train notebook

train.ipynb

@@ -84,7 +84,6 @@
     "\n",
     "import torch\n",
     "import torch.nn as nn\n",
-    "import torch.nn.functional as F\n",
     "from torchmetrics.functional import dice, jaccard_index, accuracy\n",
     "\n",
     "from segmentation_models_pytorch.losses import DiceLoss, TverskyLoss, FocalLoss\n",
@@ -94,64 +93,6 @@
     "from UNET_perso import *"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-08-28T12:57:44.214356Z",
-     "iopub.status.busy": "2023-08-28T12:57:44.213937Z",
-     "iopub.status.idle": "2023-08-28T12:57:47.351664Z",
-     "shell.execute_reply": "2023-08-28T12:57:47.350884Z",
-     "shell.execute_reply.started": "2023-08-28T12:57:44.214334Z"
-    },
-    "tags": []
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Requirement already satisfied: torchmetrics in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (1.1.0)\n",
-      "Requirement already satisfied: segmentation_models_pytorch in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (0.3.3)\n",
-      "Requirement already satisfied: scikit-image in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (0.21.0)\n",
-      "Requirement already satisfied: medpy in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (0.4.0)\n",
-      "Requirement already satisfied: numpy>1.20.0 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from torchmetrics) (1.21.6)\n",
-      "Requirement already satisfied: torch>=1.8.1 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from torchmetrics) (1.11.0)\n",
-      "Requirement already satisfied: packaging in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from torchmetrics) (23.1)\n",
-      "Requirement already satisfied: lightning-utilities>=0.8.0 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from torchmetrics) (0.9.0)\n",
-      "Requirement already satisfied: torchvision>=0.5.0 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from segmentation_models_pytorch) (0.12.0)\n",
-      "Requirement already satisfied: pretrainedmodels==0.7.4 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from segmentation_models_pytorch) (0.7.4)\n",
-      "Requirement already satisfied: efficientnet-pytorch==0.7.1 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from segmentation_models_pytorch) (0.7.1)\n",
-      "Requirement already satisfied: timm==0.9.2 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from segmentation_models_pytorch) (0.9.2)\n",
-      "Requirement already satisfied: tqdm in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from segmentation_models_pytorch) (4.65.0)\n",
-      "Requirement already satisfied: pillow in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from segmentation_models_pytorch) (9.1.1)\n",
-      "Requirement already satisfied: munch in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from pretrainedmodels==0.7.4->segmentation_models_pytorch) (4.0.0)\n",
-      "Requirement already satisfied: pyyaml in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from timm==0.9.2->segmentation_models_pytorch) (6.0)\n",
-      "Requirement already satisfied: huggingface-hub in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from timm==0.9.2->segmentation_models_pytorch) (0.16.4)\n",
-      "Requirement already satisfied: safetensors in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from timm==0.9.2->segmentation_models_pytorch) (0.3.3)\n",
-      "Requirement already satisfied: scipy>=1.8 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from scikit-image) (1.11.1)\n",
-      "Requirement already satisfied: networkx>=2.8 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from scikit-image) (3.1)\n",
-      "Requirement already satisfied: imageio>=2.27 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from scikit-image) (2.31.2)\n",
-      "Requirement already satisfied: tifffile>=2022.8.12 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from scikit-image) (2023.8.25)\n",
-      "Requirement already satisfied: PyWavelets>=1.1.1 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from scikit-image) (1.4.1)\n",
-      "Requirement already satisfied: lazy_loader>=0.2 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from scikit-image) (0.3)\n",
-      "Requirement already satisfied: SimpleITK>=1.1.0 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from medpy) (2.2.1)\n",
-      "Requirement already satisfied: typing-extensions in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from lightning-utilities>=0.8.0->torchmetrics) (4.7.1)\n",
-      "Requirement already satisfied: requests in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from torchvision>=0.5.0->segmentation_models_pytorch) (2.31.0)\n",
-      "Requirement already satisfied: filelock in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from huggingface-hub->timm==0.9.2->segmentation_models_pytorch) (3.12.2)\n",
-      "Requirement already satisfied: fsspec in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from huggingface-hub->timm==0.9.2->segmentation_models_pytorch) (2022.3.0)\n",
-      "Requirement already satisfied: charset-normalizer<4,>=2 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from requests->torchvision>=0.5.0->segmentation_models_pytorch) (2.0.12)\n",
-      "Requirement already satisfied: idna<4,>=2.5 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from requests->torchvision>=0.5.0->segmentation_models_pytorch) (3.4)\n",
-      "Requirement already satisfied: urllib3<3,>=1.21.1 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from requests->torchvision>=0.5.0->segmentation_models_pytorch) (1.26.15)\n",
-      "Requirement already satisfied: certifi>=2017.4.17 in /opt/saturncloud/envs/saturn/lib/python3.9/site-packages (from requests->torchvision>=0.5.0->segmentation_models_pytorch) (2023.7.22)\n"
-     ]
-    }
-   ],
-   "source": [
-    "!pip install torchmetrics segmentation_models_pytorch scikit-image medpy"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 4,