"examples/vscode:/vscode.git/clone" did not exist on "208df208bf2b258677ece94e86c7e6e08ad2cd41"
  1. 14 Jun, 2021 3 commits
  2. 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
  3. 10 Jun, 2021 7 commits
  4. 09 Jun, 2021 5 commits
  5. 08 Jun, 2021 6 commits
  6. 03 Jun, 2021 2 commits
  7. 02 Jun, 2021 1 commit
  8. 01 Jun, 2021 2 commits
  9. 31 May, 2021 2 commits
  10. 26 May, 2021 1 commit
  11. 25 May, 2021 4 commits
  12. 24 May, 2021 1 commit
  13. 21 May, 2021 3 commits
  14. 20 May, 2021 1 commit
  15. 19 May, 2021 1 commit