diff --git a/docs/python/how-to-guides/use-pytorch.md b/docs/python/how-to-guides/use-pytorch.md index f3f65f14ea1..6399121866e 100644 --- a/docs/python/how-to-guides/use-pytorch.md +++ b/docs/python/how-to-guides/use-pytorch.md @@ -97,7 +97,7 @@ class IrisANN(nn.Module): model = IrisANN() ``` -Now comes the fun bit: training our network. We'll calculate loss using [cross entropy](https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html), and we'll optimize the network with the [Adam algorithm](https://pytorch.org/docs/stable/generated/torch.optim.Adam.html#torch.optim.Adam). +Now comes the fun bit: training our network. We'll calculate loss using [cross entropy](https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html), and we'll optimize the network with the [Adam algorithm](https://pytorch.org/docs/stable/generated/torch.optim.Adam.html#adam). ```python test-set=1 order=null skip-test # Set our loss computation and optimization algorithms