pytorch
94f88b34 - Forward fix for DataLoader to accept custom Sharding DataPipe (#97287) (#98523)

Commit
2 years ago
Forward fix for DataLoader to accept custom Sharding DataPipe (#97287) (#98523) Fixes #96975 Changes: - Make sure custom ShardingDataPipe with `apply_sharding` can be used by `DataLoader` - Allow the `apply_sharding` function without the last argument of `sharding_group` - Make `DataLoader` not relying on `sharding_group` Pull Request resolved: https://github.com/pytorch/pytorch/pull/97287 Approved by: https://github.com/NivekT Co-authored-by: erjia <erjia@fb.com>
Author
Parents
Loading