huggingface/accelerate

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

Fechada

#2.975 aberto em 29 de jul. de 2024

 (5 comentários) (0 reação) (0 responsável)Python (626 forks)batch import
good first issue

Métricas do repositório

Stars
 (5.805 estrelas)
Métricas de merge de PR
 (Mesclagem média 16d 17h) (23 fundiu PRs em 30d)

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

Guia do colaborador