transformers
2a606f99 - Make data shuffling in `run_clm_flax.py` respect global seed (#13410)

Commit

4 years ago

Make data shuffling in `run_clm_flax.py` respect global seed (#13410) * use jax and jnp instead of numpy in data_loader * return batches as np.ndarray

References

Author

bminixhofer

Parents

transformers 2a606f99 - Make data shuffling in `run_clm_flax.py` respect global seed (#13410)

transformers
2a606f99 - Make data shuffling in `run_clm_flax.py` respect global seed (#13410)