|
| 1 | +Training references |
| 2 | +=================== |
| 3 | + |
| 4 | +On top of the many models, datasets, and image transforms, Torchvision also |
| 5 | +provides training reference scripts. These are the scripts that we use to train |
| 6 | +the :ref:`models <models>` which are then available with pre-trained weights. |
| 7 | + |
| 8 | +These scripts are not part of the core package and are instead available `on |
| 9 | +GitHub <https://github.com/pytorch/vision/tree/main/references>`_. We currently |
| 10 | +provide references for |
| 11 | +`classification <https://github.com/pytorch/vision/tree/main/references/classification>`_, |
| 12 | +`detection <https://github.com/pytorch/vision/tree/main/references/detection>`_, |
| 13 | +`segmentation <https://github.com/pytorch/vision/tree/main/references/segmentation>`_, |
| 14 | +`similarity learning <https://github.com/pytorch/vision/tree/main/references/similarity>`_, |
| 15 | +and `video classification <https://github.com/pytorch/vision/tree/main/references/video_classification>`_. |
| 16 | + |
| 17 | +While these scripts are largely stable, they do not offer backward compatibility |
| 18 | +guarantees. |
| 19 | + |
| 20 | +In general, these scripts rely on the latest (not yet released) pytorch version |
| 21 | +or the latest torchvision version. This means that to use them, **you might need |
| 22 | +to install the latest pytorch and torchvision versions**, with e.g.:: |
| 23 | + |
| 24 | + conda install pytorch torchvision -c pytorch-nightly |
| 25 | + |
| 26 | +If you need to rely on an older stable version of pytorch or torchvision, e.g. |
| 27 | +torchvision 0.10, then it's safer to use the scripts from that corresponding |
| 28 | +release on GitHub, namely |
| 29 | +https://github.com/pytorch/vision/tree/v0.10.0/references. |
0 commit comments