--train_batch_size=1 \
--gradient_accumulation_steps=4 \
--push_to_hub
```  
</hfoption>
<hfoption id="Flax">  
With Flax, you can [profile your code](https://jax.readthedocs.io/en/latest/profiling.html) by adding the `--profile_steps==5` parameter to your training command. Install the Tensorboard profile plugin:  
```bash
pip install tensorflow tensorboard-plugin-profile
tensorboard --logdir runs/fill-circle-100steps-20230411_165612/
```