Skip to content

GitLab

Unverified Commit a0c79b39 authored by Anner's avatar Anner Committed by GitHub
Browse files

Just a typo (#8341)

parent 480eec29
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
gallery/transforms/plot_transforms_e2e.py
View file @ a0c79b39
...@@ -151,8 +151,8 @@ data_loader = torch.utils.data.DataLoader( ...@@ -151,8 +151,8 @@ data_loader = torch.utils.data.DataLoader(
# We need a custom collation function here, since the object detection # We need a custom collation function here, since the object detection
# models expect a sequence of images and target dictionaries. The default # models expect a sequence of images and target dictionaries. The default
# collation function tries to torch.stack() the individual elements, # collation function tries to torch.stack() the individual elements,
# which fails in general for object detection, because the number of bouding # which fails in general for object detection, because the number of bounding
# boxes varies between the images of a same batch. # boxes varies between the images of the same batch.
collate_fn=lambda batch: tuple(zip(*batch)), collate_fn=lambda batch: tuple(zip(*batch)),
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment