HuggingFace_transformer

Files

Yu Chin Fabian Lim 4df5b9b4b2 Allow GradientAccumulationPlugin to be configured from AcceleratorConfig (#29589 )

* add gradient_accumulation_kwargs to AcceleratorConfig

* add suggestions from @muellerzr to docstrings, new behavior and tests

* Documentation suggestions from @muellerz

Co-authored-by: Zach Mueller <muellerzr@gmail.com>

* addressed @muellerzr comments regarding tests and test utils

* moved accelerate version to top of file.

* @muellerzr's variable fix

Co-authored-by: Zach Mueller <muellerzr@gmail.com>

* address @amyeroberts. fix tests and docstrings

* address @amyeroberts additional suggestions

---------

Co-authored-by: Yu Chin Fabian Lim <flim@sg.ibm.com>
Co-authored-by: Zach Mueller <muellerzr@gmail.com>

2024-03-28 14:01:40 +00:00

__init__.py

…

test_data_collator.py

handle numpy inputs in whole word mask data collator (#22032 )

2023-03-10 10:50:29 -05:00

test_trainer_callback.py

Apply ruff flake8-comprehensions (#21694 )

2023-02-22 09:14:54 +01:00

test_trainer_distributed.py

🚨 Fully revert atomic checkpointing 🚨 (#29370 )

2024-03-04 06:17:42 -05:00