pytorch/torch/utils/data
Luca Arnaboldi c3bb174bb2 SubsetRandomSampler - changed iteration over tensor to iteration over list (#149126)
Digging further the problem at https://github.com/UKPLab/sentence-transformers/pull/3261, it boils down to this expensive loop over a torch tensor. Looping over a list, like in RandomSampler, solves the issue.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149126
Approved by: https://github.com/divyanshk, https://github.com/cyyever
2025-03-31 04:33:35 +00:00
..
_utils PEP585 update - torch/utils (#145201) 2025-01-21 21:04:10 +00:00
datapipes [BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408) 2025-02-04 19:07:04 +00:00
__init__.py [BE] enable UFMT in torch.utils.data (#127705) 2024-06-27 23:16:24 +00:00
backward_compatibility.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
dataframes_pipes.ipynb
dataloader.py Move get accelerator to use build time flags when possible (#146098) 2025-03-10 13:17:58 +00:00
dataset.py Update ruff linter for PEP585 (#147540) 2025-02-22 04:45:17 +00:00
distributed.py PEP585 update - torch/utils (#145201) 2025-01-21 21:04:10 +00:00
graph_settings.py PEP585 update - torch/utils (#145201) 2025-01-21 21:04:10 +00:00
graph.py PEP585 update - torch/utils (#145201) 2025-01-21 21:04:10 +00:00
sampler.py SubsetRandomSampler - changed iteration over tensor to iteration over list (#149126) 2025-03-31 04:33:35 +00:00
standard_pipes.ipynb
typing.ipynb [BE][CI] bump ruff to 0.8.4 (#143753) 2024-12-24 12:24:10 +00:00