File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -12,7 +12,7 @@ class VisionDataset(data.Dataset):
12
12
It is necessary to override the ``__getitem__`` and ``__len__`` method.
13
13
14
14
Args:
15
- root (string): Root directory of dataset.
15
+ root (string, optional ): Root directory of dataset. Only used for `__repr__` .
16
16
transforms (callable, optional): A function/transforms that takes in
17
17
an image and a label and returns the transformed versions of both.
18
18
transform (callable, optional): A function/transform that takes in an PIL image
@@ -29,7 +29,7 @@ class VisionDataset(data.Dataset):
29
29
30
30
def __init__ (
31
31
self ,
32
- root : str ,
32
+ root : Optional [ str ] = None ,
33
33
transforms : Optional [Callable ] = None ,
34
34
transform : Optional [Callable ] = None ,
35
35
target_transform : Optional [Callable ] = None ,
You can’t perform that action at this time.
0 commit comments