search

pytorch / data

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
BSD 3-Clause "New" or "Revised" License
1.08k stars 142 forks source link

fork dataloader __init__ instead of patching samplers #1281

Closed andrewkho closed 2 days ago

andrewkho commented 3 days ago

Fixes #1280

Currently we patch torch.utils.data's BatchSampler and RandomSampler in order to add statefulness to them. This is way too aggressive.

Instead let's just fork the dataloader init for now and merge back into core once the time is right.

Changes

pytorch-bot[bot] commented 3 days ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/data/1281

Note: Links to docs will display an error until the docs builds have been completed.

:heavy_exclamation_mark: 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

:white_check_mark: No Failures

As of commit c8017ed033eabb16180a8824fc3f563b4894a787 with merge base b0e25e26b6cbe78c7c26c3c8d1acb61a30a3def9 (image): :green_heart: Looks good so far! There are no failures yet. :green_heart:

This comment was automatically generated by Dr. CI and updates every 15 minutes.