huggingface/accelerate

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

Chiusa

#2975 aperta il 29 lug 2024

 (5 commenti) (0 reazioni) (0 assegnatari)Python (626 fork)batch import
good first issue

Metriche repository

Star
 (5805 stelle)
Metriche merge PR
 (Merge medio 16g 17h) (23 PR mergiate in 30 g)

Descrizione

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

Guida contributor