Skip to content

Commit 228af50

Browse files
prabhat00155facebook-github-bot
authored andcommitted
[fbsync] Added docs for reference scripts (#4574)
Reviewed By: NicolasHug Differential Revision: D31505554 fbshipit-source-id: 2bacf2d55bba32a26878a35c04af872be51a0bf3
1 parent 2589a79 commit 228af50

File tree

3 files changed

+33
-1
lines changed

3 files changed

+33
-1
lines changed

docs/source/index.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,10 @@ architectures, and common image transformations for computer vision.
4141

4242
.. toctree::
4343
:maxdepth: 1
44-
:caption: Examples
44+
:caption: Examples and training references
4545

4646
auto_examples/index
47+
training_references
4748

4849
.. automodule:: torchvision
4950
:members:

docs/source/models.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
.. _models:
2+
13
torchvision.models
24
##################
35

docs/source/training_references.rst

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
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

Comments
 (0)