Add `join_uneven_inputs` context manager to Accelerator #820
muellerzr
merged
19 commits into
huggingface:main
from
ch/join_context_manager
Add test for join context manager
80420e52
Add join_uneven_inputs context manager
71aa7df8
Format
bc4d6ac1
add conditional import for join
02a89e30
Replace bare yield with nullcontext
dc028e6f
sgugger
commented
on 2022-11-04
pacman100
commented
on 2022-11-07
Update accelerator to maintain references to dataloaders
bff03f4b
add override option to join context manager
87f2db6a
format
3da680c7
Add minimal docstring
124057ca
updates based on initial feedback
05543103
remove launcher used for local testing from test script
54f690a6
fix quality issues
129153b5
Merge branch 'main' into ch/join_context_manager
9c60f67c
DEBUG: try resetting accelerator state to fix test
a13a56ea
Revert "DEBUG: try resetting accelerator state to fix test"
e6acb1c1
Reset state after accelerator tests
93521577
sgugger
approved these changes
on 2022-11-10
muellerzr
approved these changes
on 2022-11-10
Update src/accelerate/accelerator.py
f26b9ca9
Warn if at least one iterable dataset seen
71341c43
remove launcher used for local test running
bd391ada
Chris-hughes10
changed the title
Login to write a write a comment.
Login via GitHub