Skip to content

Commit

Permalink
Commit from GitHub Actions (Build Notebooks)
Browse files Browse the repository at this point in the history
  • Loading branch information
Ben-Salmon committed Aug 25, 2024
1 parent 5cb799f commit 3ea2cbe
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 7 deletions.
11 changes: 5 additions & 6 deletions 01_CARE/exercise.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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)"
]
},
{
Expand Down Expand Up @@ -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",
"</div>"
]
Expand All @@ -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",
Expand Down
2 changes: 1 addition & 1 deletion 04_DenoiSplit/exercise.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -796,7 +796,7 @@
},
{
"cell_type": "markdown",
"id": "2c60515d",
"id": "37ef9a4b",
"metadata": {},
"source": [
"<hr style=\"height:2px;\"><div class=\"alert alert-block alert-success\"><h1>End of the exercise</h1>\n",
Expand Down

0 comments on commit 3ea2cbe

Please sign in to comment.