Unverified Commit 127e8a70 authored by Zeeshan Khan Suri's avatar Zeeshan Khan Suri Committed by GitHub
Browse files

Making root parameter optional for Vision Dataset (#8124)


Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
Co-authored-by: default avatarNicolas Hug <nh.nicolas.hug@gmail.com>
parent c7e29470
...@@ -12,7 +12,7 @@ class VisionDataset(data.Dataset): ...@@ -12,7 +12,7 @@ class VisionDataset(data.Dataset):
It is necessary to override the ``__getitem__`` and ``__len__`` method. It is necessary to override the ``__getitem__`` and ``__len__`` method.
Args: Args:
root (string): Root directory of dataset. root (string, optional): Root directory of dataset. Only used for `__repr__`.
transforms (callable, optional): A function/transforms that takes in transforms (callable, optional): A function/transforms that takes in
an image and a label and returns the transformed versions of both. an image and a label and returns the transformed versions of both.
transform (callable, optional): A function/transform that takes in an PIL image transform (callable, optional): A function/transform that takes in an PIL image
...@@ -29,7 +29,7 @@ class VisionDataset(data.Dataset): ...@@ -29,7 +29,7 @@ class VisionDataset(data.Dataset):
def __init__( def __init__(
self, self,
root: str, root: Optional[str] = None,
transforms: Optional[Callable] = None, transforms: Optional[Callable] = None,
transform: Optional[Callable] = None, transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None, target_transform: Optional[Callable] = None,
......
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