huggingface/accelerate

Barebones dataloader to allow for any type of iterable dataloader-like object to be used. Should just handle device placement

Fermée

#2 975 ouverte le 29 juil. 2024

 (5 commentaires) (0 réaction) (0 personne assignée)Python (626 forks)batch import
good first issue

Métriques du dépôt

Stars
 (5 805 étoiles)
Métriques de merge PR
 (Merge moyen 16j 17h) (23 PRs mergées en 30 j)

Description

Make a simplistic version of the DispatchDataLoader which allows for users to easily pass in any Iterable type of object and call it's __iter__. Likely situation to make sure Accelerator knows what's up during prepare is to add an option to the DataLoaderConfiguration for custom_classes which accepts types for us to do isinstance() on

Guide contributeur