# Training arguments for flax

**URL:** <https://discuss.huggingface.co/t/training-arguments-for-flax/15696>\
**Category:** 🤗Transformers\
**Created:** [March 14, 2022, 3:10pm UTC](https://discuss.huggingface.co/t/training-arguments-for-flax/15696 "2022-03-14T15:10:10Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![mrdrozdov](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/mrdrozdov/32/2692_2.png) [@mrdrozdov](https://discuss.huggingface.co/u/mrdrozdov)\
**Post date:** [March 14, 2022, 3:10pm UTC](https://discuss.huggingface.co/t/training-arguments-for-flax/15696/1 "2022-03-14T15:10:10Z")

</div>

Noticed there is no gradient accumulation option for the flax text classification example.

The language modeling example uses TrainingArguments, which includes gradient accumulation and other options. Is there any plan to update the text classification example?