Revert D15511921: [pytorch][PR] BatchSampler now uses list.clear() in…

…stead of creating new objects Differential Revision: D15511921 Original commit changeset: e943d21e75e1 fbshipit-source-id: 933b7ef74c7a530f0a2cc087c8ee6f0455cf9239
lcy-seso · May 27, 2019 · 6480d3f · 6480d3f
1 parent 482ae8e
commit 6480d3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/utils/data/sampler.py b/torch/utils/data/sampler.py
@@ -173,7 +173,7 @@ def __iter__(self):
             batch.append(idx)
             if len(batch) == self.batch_size:
                 yield batch
-                batch.clear()
+                batch = []
         if len(batch) > 0 and not self.drop_last:
             yield batch