huggingface/accelerate

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

已關閉

#2,975 建立於 2024年7月29日

在 GitHub 查看貢獻者指南
 (5 則留言) (0 個反應) (0 位負責人)Python (626 個分叉)batch import
good first issue

倉庫指標

星標
 (5,805 顆星)
PR 合併指標
 (平均合併 16天 17小時) (30 天內合併 23 個 PR)

描述

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

貢獻者指南

研究方向
閱讀DispatchDataLoader的原始碼以了解其結構。找到DataLoaderConfiguration定義的位置並修改它以接受custom classes。建立一個繼承自Iterable並實作 iter 的新類別，根據需要將資料移動到裝置上。最後，更新Accelerator.prepare以包含對新類別的isinstance檢查。
技術棧
pythonpytorch
領域
backend
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 小時
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
GitPythonPyTorch
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
75