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

bug fix

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