1. 19 Jan, 2022 1 commit
    • Kamal Raj's avatar
      [FLAX] glue training example refactor (#13815) · d1f5ca1a
      Kamal Raj authored
      * refactor run_flax_glue.py
      
      * updated readme
      
      * rm unused import and args typo fix
      
      * refactor
      
      * make consistent arg name across task
      
      * has_tensorboard check
      
      * argparse -> argument dataclasses
      
      * refactor according to review
      
      * fix
      d1f5ca1a
  2. 12 Dec, 2021 1 commit
  3. 06 Dec, 2021 2 commits
    • Suraj Patil's avatar
      fix flax examples tests (#14646) · 75ae287a
      Suraj Patil authored
      * make tensorboard optional
      
      * update test_fetcher for flax examples
      
      * make the tests slow
      75ae287a
    • Suraj Patil's avatar
      Add Flax example tests (#14599) · c5bd732a
      Suraj Patil authored
      * add test for glue
      
      * add tests for clm
      
      * fix clm test
      
      * add summrization tests
      
      * more tests
      
      * fix few tests
      
      * add test for t5 mlm
      
      * fix t5 mlm test
      
      * fix tests for multi device
      
      * cleanup
      
      * ci job
      
      * fix metric file name
      
      * make t5 more robust
      c5bd732a