Skip to content

Commit 8837e0e

Browse files
zhiqwangfmassa
authored andcommitted
Remove duplicate code (#1125)
* Remove duplicate code * Passed transform and target_transform to the constructor
1 parent 8d580a1 commit 8837e0e

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

torchvision/datasets/fakedata.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,11 @@ class FakeData(VisionDataset):
2121

2222
def __init__(self, size=1000, image_size=(3, 224, 224), num_classes=10,
2323
transform=None, target_transform=None, random_offset=0):
24-
super(FakeData, self).__init__(None)
25-
self.transform = transform
26-
self.target_transform = target_transform
24+
super(FakeData, self).__init__(None, transform=transform,
25+
target_transform=target_transform)
2726
self.size = size
2827
self.num_classes = num_classes
2928
self.image_size = image_size
30-
self.transform = transform
31-
self.target_transform = target_transform
3229
self.random_offset = random_offset
3330

3431
def __getitem__(self, index):

0 commit comments

Comments
 (0)