1. 30 Sep, 2021 1 commit
    • Suraj Patil's avatar
      [examples/flax] use Repository API for push_to_hub (#13672) · 7db2a79b
      Suraj Patil authored
      * use Repository for push_to_hub
      
      * update readme
      
      * update other flax scripts
      
      * update readme
      
      * update qa example
      
      * fix push_to_hub call
      
      * fix typo
      
      * fix more typos
      
      * update readme
      
      * use abosolute path to get repo name
      
      * fix glue script
      7db2a79b
  2. 28 Aug, 2021 1 commit
  3. 27 Aug, 2021 1 commit
  4. 09 Aug, 2021 1 commit
  5. 27 Jul, 2021 1 commit
  6. 20 Jul, 2021 1 commit
  7. 09 Jul, 2021 1 commit
  8. 08 Jul, 2021 1 commit
  9. 07 Jul, 2021 1 commit
  10. 06 Jul, 2021 1 commit
  11. 05 Jul, 2021 2 commits
  12. 29 Jun, 2021 1 commit
  13. 28 Jun, 2021 1 commit
  14. 25 Jun, 2021 1 commit
  15. 11 Jun, 2021 1 commit
    • Suraj Patil's avatar
      Flax CLM script (#12023) · 15b498f3
      Suraj Patil authored
      * first draft
      
      * max_seq_length => block_size
      
      * fix arg names
      
      * fix typos
      
      * fix loss calculation
      
      * add max examples, fix  train eval steps, metrics
      
      * optimizer mask
      
      * fix perpelexity, metric logging
      
      * fix logging
      
      * data_collator = > data_loader
      
      * refactor loss_fn
      
      * support single GPU
      
      * pass distributed to write_metric
      
      * fix jitting
      
      * fix single device training
      
      * fix single device metrics
      
      * close inner progress bars once finished
      
      * add overwrite_cache arg
      
      * ifx dataset caching issue
      
      * add more logs
      
      * few small fixes,
      
      * address nicholas suggestions
      
      * fix docstr
      
      * address patricks suggestions
      
      * make flake happy
      
      * pass new new_dropout_rng to apply_gradients
      
      * reset train metrics after every epoc
      
      * remove distributed logis, small fixes
      15b498f3
  16. 08 Jun, 2021 1 commit
  17. 25 May, 2021 3 commits
  18. 12 May, 2021 2 commits
  19. 11 May, 2021 1 commit
  20. 29 Apr, 2021 1 commit
  21. 26 Apr, 2021 1 commit
  22. 23 Apr, 2021 1 commit
  23. 21 Apr, 2021 1 commit
  24. 13 Apr, 2021 1 commit
  25. 09 Apr, 2021 1 commit
  26. 08 Apr, 2021 1 commit
  27. 07 Apr, 2021 2 commits
  28. 06 Apr, 2021 2 commits
  29. 31 Mar, 2021 1 commit
  30. 16 Mar, 2021 2 commits
  31. 15 Mar, 2021 1 commit
  32. 08 Mar, 2021 1 commit
  33. 27 Feb, 2021 1 commit