Refactor BucketBatch (#63185)
authorErjia Guan <erjia@fb.com>
Mon, 16 Aug 2021 13:39:56 +0000 (06:39 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 16 Aug 2021 13:42:56 +0000 (06:42 -0700)
commitd1cbee7b2b1da94ff1a0fa880bd0173de27fb89f
tree202e8212756a4fc5c843565d0994d8a8a5b045b0
parent56d609d93e811c18ab710e8b6f4fd9836539ba01
Refactor BucketBatch (#63185)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63185

Test Plan: Imported from OSS

Reviewed By: bdhirsh

Differential Revision: D30288893

Pulled By: ejguan

fbshipit-source-id: b88b792d12a83c99d8ea9e516e3b4c54a82100f6
test/test_datapipe.py
torch/utils/data/datapipes/iter/__init__.py
torch/utils/data/datapipes/iter/grouping.py