Skip to content

Commit 092909a

Browse files
committed
[skip-ci] Few more rendering fixes
1 parent a2b39cb commit 092909a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

intermediate_source/torchvision_tutorial.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -356,6 +356,7 @@ def get_model_instance_segmentation(num_classes):
356356
os.system("wget https://raw.githubusercontent.com/pytorch/vision/main/references/detection/coco_eval.py")
357357
os.system("wget https://raw.githubusercontent.com/pytorch/vision/main/references/detection/transforms.py")
358358

359+
######################################################################
359360
# Since v0.15.0 torchvision provides `new Transforms API <https://pytorch.org/vision/stable/transforms.html>`_
360361
# to easily write data augmentation pipelines for Object Detection and Segmentation tasks.
361362
#
@@ -373,7 +374,7 @@ def get_transform(train):
373374
transforms.append(T.ToPureTensor())
374375
return T.Compose(transforms)
375376

376-
377+
######################################################################
377378
# Testing ``forward()`` method (Optional)
378379
# ---------------------------------------
379380
#

0 commit comments

Comments
 (0)