From 3ea2cbe95166d47a8f9662bdb2fb321a04082cea Mon Sep 17 00:00:00 2001 From: Ben-Salmon Date: Sun, 25 Aug 2024 11:42:31 +0000 Subject: [PATCH] Commit from GitHub Actions (Build Notebooks) --- 01_CARE/exercise.ipynb | 11 +++++------ 04_DenoiSplit/exercise.ipynb | 2 +- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/01_CARE/exercise.ipynb b/01_CARE/exercise.ipynb index 5f5d67c..026750e 100644 --- a/01_CARE/exercise.ipynb +++ b/01_CARE/exercise.ipynb @@ -647,7 +647,7 @@ "outputs": [], "source": [ "# Load the model\n", - "model = UNet(depth=3, in_channels=1, out_channels=1)" + "model = UNet(depth=2, in_channels=1, out_channels=1)" ] }, { @@ -746,10 +746,9 @@ "![image](nb_data/extensions.png)\n", "\n", "2) Search Tensorboard and install and install the extension published by Microsoft.\n", - "3) Open the command palette (ctrl+shift+p), search for Python: Launch Tensorboard and hit enter.\n", - "4) When prompted, select either \"Use current working directory\" or \"Select another folder\" and enter the path to the `01_CARE/runs/` directory.\n", - "\n", - "Once it's open, continue to the next cell to start training, then open the Tensorboard tab to see the loss curves.\n", + "3) Start training. Run the cell below to begin training the model and generating logs.\n", + "3) Once training is started. Open the command palette (ctrl+shift+p), search for Python: Launch Tensorboard and hit enter.\n", + "4) When prompted, select \"Select another folder\" and enter the path to the `01_CARE/runs/` directory.\n", "\n", "" ] @@ -772,7 +771,7 @@ "outputs": [], "source": [ "# Training loop\n", - "n_epochs = 10\n", + "n_epochs = 5\n", "device = \"cuda\" if cuda.is_available() else \"cpu\"\n", "model.to(device)\n", "\n", diff --git a/04_DenoiSplit/exercise.ipynb b/04_DenoiSplit/exercise.ipynb index 76436d0..65b7aab 100644 --- a/04_DenoiSplit/exercise.ipynb +++ b/04_DenoiSplit/exercise.ipynb @@ -796,7 +796,7 @@ }, { "cell_type": "markdown", - "id": "2c60515d", + "id": "37ef9a4b", "metadata": {}, "source": [ "

End of the exercise

\n",