Skip to content

Commit 6f8606e

Browse files
minor changes
1 parent 53c85b9 commit 6f8606e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/torchvision_tutorial.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ If your model returns the above methods, they will make it work for both
5656
training and evaluation, and will use the evaluation scripts from
5757
``pycocotools``.
5858

59-
One note on the ``labels``. The model expects class ``0`` to be always the background. If your dataset does not contain the background class, you should not have ``0`` in your ``labels``. For example, assuming you have just two classes, *cat* and *dog*, you can define ``1`` (not ``0``) to represent *cats* and ``2`` to represent *dogs*. If in your image you have booth classes, your ``labels`` tensor should look like ``[1,2]``.
59+
One note on the ``labels``. The model considers class ``0`` as background. If your dataset does not contain the background class, you should not have ``0`` in your ``labels``. For example, assuming you have just two classes, *cat* and *dog*, you can define ``1`` (not ``0``) to represent *cats* and ``2`` to represent *dogs*. So, for instance, if one of the images has booth classes, your ``labels`` tensor should look like ``[1,2]``.
6060

6161
Additionally, if you want to use aspect ratio grouping during training
6262
(so that each batch only contains images with similar aspect ratio),

0 commit comments

Comments
 (0)