Skip to content

Commit c11858f

Browse files
committed
fix checking target_type argument values
1 parent fb846a2 commit c11858f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/datasets/widerface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def __init__(
8080
else:
8181
self.target_type = [target_type]
8282
self.target_type = [verify_str_arg(t, "target_type", ("raw", "bbox", "attr", ""))
83-
for t in target_type]
83+
for t in self.target_type]
8484

8585
if not self.target_type and self.target_transform is not None:
8686
raise RuntimeError('target_transform is specified but target_type is empty')

0 commit comments

Comments
 (0)