accelerate
5d558f21 - [WIP] Implementing gather_for_metrics with dedup for non tensor objects (#1937)

Commit
2 years ago
[WIP] Implementing gather_for_metrics with dedup for non tensor objects (#1937) * [feat] implementing gather_for_metrics for objects * [lint] make style result * [docs] improve fn docs gather for metrics Co-authored-by: Zach Mueller <muellerzr@gmail.com> * [docs] update args description gather for metrics Co-authored-by: Zach Mueller <muellerzr@gmail.com> * [refactor] gather for metrics for non tensor obj Co-authored-by: Zach Mueller <muellerzr@gmail.com> * [fix] renaming tensor to data (was not defined and it is not just a tensor) * [fix] else state * [test] gather for metrics with non tensor objects * [lint] make style result * Update src/accelerate/accelerator.py Co-authored-by: Zach Mueller <muellerzr@gmail.com> * Update src/accelerate/accelerator.py Co-authored-by: Zach Mueller <muellerzr@gmail.com> * [test] removing useless assertion Co-authored-by: Zach Mueller <muellerzr@gmail.com> * [test] add running on main * [lint] style autoformat --------- Co-authored-by: Lorenzobattistela <lorenzobattistela@gmail.com> Co-authored-by: Zach Mueller <muellerzr@gmail.com>
Parents
Loading