Skip to content

GitLab

Unverified Commit 3b0b201a authored by Shuaizhen Jing's avatar Shuaizhen Jing Committed by GitHub
Browse files

add @torch.jit.script to ImageList (#1919)

parent e1e975f9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
torchvision/models/detection/image_list.py
View file @ 3b0b201a
......@@ -6,6 +6,7 @@ from torch.jit.annotations import List, Tuple
from torch import Tensor
@torch.jit.script
class ImageList(object):
"""
Structure that holds a list of images (of possibly
......
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