roberta-swedish / run_mlm_flax_stream.py
birgermoell's picture
Saving weights and logs of step 1001
b3b986d
raw
history blame contribute delete
106 Bytes
/home/bmoell/transformers/examples/research_projects/jax-projects/dataset-streaming/run_mlm_flax_stream.py