Introduce AcceleratorConfig dataclass #28664
muellerzr
merged
18 commits into
main
from
muellerzr-accelerator-dataclass
amyeroberts
approved these changes
on 2024-02-14
Introduce acceleratorconfig dataclass
f8288ecc
Extra second warn
b2fc85b9
Move import
28fe8e03
Try moving import under is_accelerate_available
5d9d7311
Quality
7d186f6c
Apply suggestions from code review
aa539099
Clean
2f6f1253
Remove to_kwargs
1756ef53
Change version
7a19b33a
Improve tests by including dispatch and split batches
464db0c5
Improve reliability
5b9c318e
Update tests/trainer/test_trainer.py
0d69f6cf
Fixup tests and review nits
8cfa8d71
Make tests pass
fe964fb6
protect import
a43a29d1
Protect import
f622edd9
Empty-Commit
1de6ff51
Make training_args.to_dict handle the AcceleratorConfig
f5b42811
devCaiquedePaula
commented
on 2024-02-19
Reviewers
Assignees
No one assigned
Labels
trainer
Milestone
No milestone
Login to write a write a comment.
Login via GitHub