Skip to content

Commit

Permalink
[Cherry-pick for 0.18] Just a typo (#8341) (#8355)
Browse files Browse the repository at this point in the history
Co-authored-by: Anner <[email protected]>
Co-authored-by: Nicolas Hug <[email protected]>
  • Loading branch information
3 people authored Mar 26, 2024
1 parent 7a4bdf3 commit f22b01f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions gallery/transforms/plot_transforms_e2e.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,8 +151,8 @@
# We need a custom collation function here, since the object detection
# models expect a sequence of images and target dictionaries. The default
# collation function tries to torch.stack() the individual elements,
# which fails in general for object detection, because the number of bouding
# boxes varies between the images of a same batch.
# which fails in general for object detection, because the number of bounding
# boxes varies between the images of the same batch.
collate_fn=lambda batch: tuple(zip(*batch)),
)

Expand Down

0 comments on commit f22b01f

Please sign in to comment.