Skip to content

Commit dd904ec

Browse files
committed
fix broken link
1 parent 7e8cc71 commit dd904ec

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

project-dog-classification/dog_app.ipynb

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@
269269
"\n",
270270
"In the next code cell, you will write a function that accepts a path to an image (such as `'dogImages/train/001.Affenpinscher/Affenpinscher_00001.jpg'`) as input and returns the index corresponding to the ImageNet class that is predicted by the pre-trained VGG-16 model. The output should always be an integer between 0 and 999, inclusive.\n",
271271
"\n",
272-
"Before writing the function, make sure that you take the time to learn how to appropriately pre-process tensors for pre-trained models in the [PyTorch documentation](http://pytorch.org/docs/stable/torchvision/models.html)."
272+
"Before writing the function, make sure that you take the time to learn how to appropriately pre-process tensors for pre-trained models in the [PyTorch documentation](https://pytorch.org/vision/stable/models.html)."
273273
]
274274
},
275275
{
@@ -550,9 +550,10 @@
550550
"outputs": [],
551551
"source": [
552552
"# the following import is required for training to be robust to truncated images\n",
553-
"from PIL import ImageFile\n",
554-
"ImageFile.LOAD_TRUNCATED_IMAGES = True\n\n",
555-
"def train(n_epochs, loaders, model, optimizer, criterion, use_cuda, save_path):\n",
553+
"from PIL import ImageFile\n",
554+
"ImageFile.LOAD_TRUNCATED_IMAGES = True\n",
555+
"\n",
556+
"def train(n_epochs, loaders, model, optimizer, criterion, use_cuda, save_path):\n",
556557
" \"\"\"returns trained model\"\"\"\n",
557558
" # initialize tracker for minimum validation loss\n",
558559
" valid_loss_min = np.Inf \n",
@@ -921,4 +922,4 @@
921922
},
922923
"nbformat": 4,
923924
"nbformat_minor": 1
924-
}
925+
}

0 commit comments

Comments
 (0)