|
208 | 208 | # 5. Test loop (`test_step`) which is the same as validation, only on a test set. |
209 | 209 | # |
210 | 210 | # Therefore, we don't abstract the PyTorch code, but rather organize it and define some default operations that are commonly used. |
211 | | -# If you need to change something else in your training/validation/test loop, there are many possible functions you can overwrite (see the [docs](https://pytorch-lightning.readthedocs.io/en/stable/lightning_module.html) for details). |
| 211 | +# If you need to change something else in your training/validation/test loop, there are many possible functions you can overwrite (see the [docs](https://pytorch-lightning.readthedocs.io/en/stable/common/lightning_module.html) for details). |
212 | 212 | # |
213 | 213 | # Now we can look at an example of how a Lightning Module for training a CNN looks like: |
214 | 214 |
|
@@ -322,7 +322,7 @@ def create_model(model_name, model_hparams): |
322 | 322 | # Besides the Lightning module, the second most important module in PyTorch Lightning is the `Trainer`. |
323 | 323 | # The trainer is responsible to execute the training steps defined in the Lightning module and completes the framework. |
324 | 324 | # Similar to the Lightning module, you can override any key part that you don't want to be automated, but the default settings are often the best practice to do. |
325 | | -# For a full overview, see the [documentation](https://pytorch-lightning.readthedocs.io/en/stable/trainer.html). |
| 325 | +# For a full overview, see the [documentation](https://pytorch-lightning.readthedocs.io/en/stable/common/trainer.html). |
326 | 326 | # The most important functions we use below are: |
327 | 327 | # |
328 | 328 | # * `trainer.fit`: Takes as input a lightning module, a training dataset, and an (optional) validation dataset. |
@@ -764,7 +764,7 @@ def forward(self, x): |
764 | 764 | # |
765 | 765 | # The three groups operate on the resolutions $32\times32$, $16\times16$ and $8\times8$ respectively. |
766 | 766 | # The blocks in orange denote ResNet blocks with downsampling. |
767 | | -# The same notation is used by many other implementations such as in the [torchvision library](https://pytorch.org/docs/stable/_modules/torchvision/models/resnet.html#resnet18) from PyTorch. |
| 767 | +# The same notation is used by many other implementations such as in the [torchvision library](https://pytorch.org/vision/0.11/models.html#torchvision.models.resnet18) from PyTorch. |
768 | 768 | # Thus, our code looks as follows: |
769 | 769 |
|
770 | 770 |
|
|
0 commit comments