Skip to content

Commit 554233b

Browse files
committed
Update docs/07_pytorch_experiment_tracking.ipynb
Fix typo and tensorboard installation
1 parent 8adeffb commit 554233b

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

docs/07_pytorch_experiment_tracking.ipynb

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -725,7 +725,13 @@
725725
"metadata": {},
726726
"outputs": [],
727727
"source": [
728-
"from torch.utils.tensorboard import SummaryWriter\n",
728+
"try:\n",
729+
" from torch.utils.tensorboard import SummaryWriter\n",
730+
"except:\n",
731+
" print(\"[INFO] Couldn't find tensorboard... installing it.\")\n",
732+
" !pip install -q tensorboard\n",
733+
" from torch.utils.tensorboard import SummaryWriter\n",
734+
"\n",
729735
"\n",
730736
"# Create a writer with all default settings\n",
731737
"writer = SummaryWriter()"
@@ -2254,7 +2260,7 @@
22542260
"source": [
22552261
"Running the cell above we should get an output similar to the following.\n",
22562262
"\n",
2257-
"> **Note:** Depending on the random seeds you used/hardware you used there's a chance your numbers aren't exactly the same as what's here. This is okay. It's due to the inheret randomness of deep learning. What matters most is the trend. Where your numbers are heading. If they're off by a large amount, perhaps there's something wrong and best to go back and check the code. But if they're off by a small amount (say a couple of decimal places or so), that's okay. \n",
2263+
"> **Note:** Depending on the random seeds you used/hardware you used there's a chance your numbers aren't exactly the same as what's here. This is okay. It's due to the inherent randomness of deep learning. What matters most is the trend. Where your numbers are heading. If they're off by a large amount, perhaps there's something wrong and best to go back and check the code. But if they're off by a small amount (say a couple of decimal places or so), that's okay. \n",
22582264
"\n",
22592265
"<img src=\"https://raw.githubusercontent.com/mrdbourke/pytorch-deep-learning/main/images/07-tensorboard-lowest-test-loss.png\" alt=\"various modelling experiments visualized on tensorboard with model that has the lowest test loss highlighted\" width=900/>\n",
22602266
"\n",

0 commit comments

Comments
 (0)