1. 16 Feb, 2024 1 commit
  2. 29 Sep, 2023 1 commit
    • Sanchit Gandhi's avatar
      [Flax Examples] Seq2Seq ASR Fine-Tuning Script (#21764) · 68e85fc8
      Sanchit Gandhi authored
      * from seq2seq speech
      
      * [Flax] Example script for speech seq2seq
      
      * tests and fixes
      
      * make style
      
      * fix: label padding tokens
      
      * fix: label padding tokens over list
      
      * update ln names for Whisper
      
      * try datasets iter loader
      
      * create readme and append results
      
      * style
      
      * make style
      
      * adjust lr
      
      * use pt dataloader
      
      * make fast
      
      * pin gen max len
      
      * finish
      
      * add pt to requirements for test
      
      * fix pt -> torch
      
      * add accelerate
      68e85fc8
  3. 04 Jan, 2023 1 commit
  4. 11 Apr, 2022 1 commit
  5. 23 Feb, 2022 1 commit
  6. 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
  7. 12 Dec, 2021 1 commit
  8. 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