Commit 0f6289a4 authored by Yuxin Wu's avatar Yuxin Wu

bug fix

parent 0fa11e65
......@@ -108,9 +108,6 @@ class BatchDataByShape(BatchData):
super(BatchDataByShape, self).__init__(ds, batch_size, remainder=False)
self.idx = idx
def size(self):
raise NotImplementedError()
def reset_state(self):
super(BatchDataByShape, self).reset_state()
self.holder = defaultdict(list)
......
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