Commit 8837e0ef authored by Zhiqiang Wang's avatar Zhiqiang Wang Committed by Francisco Massa
Browse files

Remove duplicate code (#1125)

* Remove duplicate code

* Passed transform and target_transform to the constructor
parent 8d580a1f
...@@ -21,14 +21,11 @@ class FakeData(VisionDataset): ...@@ -21,14 +21,11 @@ class FakeData(VisionDataset):
def __init__(self, size=1000, image_size=(3, 224, 224), num_classes=10, def __init__(self, size=1000, image_size=(3, 224, 224), num_classes=10,
transform=None, target_transform=None, random_offset=0): transform=None, target_transform=None, random_offset=0):
super(FakeData, self).__init__(None) super(FakeData, self).__init__(None, transform=transform,
self.transform = transform target_transform=target_transform)
self.target_transform = target_transform
self.size = size self.size = size
self.num_classes = num_classes self.num_classes = num_classes
self.image_size = image_size self.image_size = image_size
self.transform = transform
self.target_transform = target_transform
self.random_offset = random_offset self.random_offset = random_offset
def __getitem__(self, index): def __getitem__(self, index):
......
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