Skip to content

Commit 8c5bbe5

Browse files
sayakpaulstevhliu
andauthored
Reword E2E training and inference tips in the vision guides (#5217)
* chore: reword e2e workflow suggestions for image guides. Co-authored-by: Steven Liu <[email protected]> * Apply suggestions from code review Co-authored-by: Steven Liu <[email protected]> Co-authored-by: Steven Liu <[email protected]>
1 parent 2945690 commit 8c5bbe5

File tree

3 files changed

+16
-6
lines changed

3 files changed

+16
-6
lines changed

docs/source/image_classification.mdx

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,8 @@ You can verify the transformation worked by indexing into the `pixel_values` of
8282

8383
<Tip>
8484

85-
To learn how to leverage 🤗 Datasets end-to-end for training an image classification model, refer to
86-
[this notebook](https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/image_classification.ipynb).
85+
Now that you know how to process a dataset for image classification, learn
86+
[how to train an image classification model](https://colab.research.google.com/github/huggingface/notebooks/blob/main/examples/image_classification.ipynb)
87+
and use it for inference.
8788

8889
</Tip>

docs/source/object_detection.mdx

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -154,9 +154,14 @@ You can verify the transform works by visualizing the 10th example:
154154
... )
155155
```
156156

157-
To learn how to leverage parts of 🤗 datasets end-to-end for training an object detection model, refer to
158-
[this notebook](https://nbviewer.org/github/NielsRogge/Transformers-Tutorials/blob/master/YOLOS/Fine_tuning_YOLOS_for_object_detection_on_custom_dataset_%28balloon%29.ipynb).
159-
160157
<div class="flex justify-center">
161158
<img src="https://huggingface.co/datasets/nateraw/documentation-images/resolve/main/visualize_detection_example_transformed_2.png">
162159
</div>
160+
161+
<Tip>
162+
163+
Now that you know how to process a dataset for object detection, learn
164+
[how to train an object detection model](https://colab.research.google.com/github/NielsRogge/Transformers-Tutorials/blob/master/YOLOS/Fine_tuning_YOLOS_for_object_detection_on_custom_dataset_(balloon).ipynb)
165+
and use it for inference.
166+
167+
</Tip>

docs/source/semantic_segmentation.mdx

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,10 @@ In this guide, you have used `albumentations` for augmenting the dataset. It's a
168168
<img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/datasets/torchvision_seg.png">
169169
</div>
170170

171+
<Tip>
172+
171173
Now that you know how to process a dataset for semantic segmentation, learn
172174
[how to train a semantic segmentation model](https://huggingface.co/docs/transformers/tasks/semantic_segmentation)
173-
and use it for inference.
175+
and use it for inference.
176+
177+
</Tip>

0 commit comments

Comments
 (0)