1. 30 Jan, 2024 1 commit
    • Matt's avatar
      Add tf_keras imports to prepare for Keras 3 (#28588) · 415e9a09
      Matt authored
      * Port core files + ESM (because ESM code is odd)
      
      * Search-replace in modelling code
      
      * Fix up transfo_xl as well
      
      * Fix other core files + tests (still need to add correct import to tests)
      
      * Fix cookiecutter
      
      * make fixup, fix imports in some more core files
      
      * Auto-add imports to tests
      
      * Cleanup, add imports to sagemaker tests
      
      * Use correct exception for importing tf_keras
      
      * Fixes in modeling_tf_utils
      
      * make fixup
      
      * Correct version parsing code
      
      * Ensure the pipeline tests correctly revert to float32 after each test
      
      * Ensure the pipeline tests correctly revert to float32 after each test
      
      * More tf.keras -> keras
      
      * Add dtype cast
      
      * Better imports of tf_keras
      
      * Add a cast for tf.assign, just in case
      
      * Fix callback imports
      415e9a09
  2. 09 May, 2023 1 commit
  3. 08 May, 2023 1 commit
  4. 01 Feb, 2023 1 commit
  5. 22 Sep, 2022 1 commit
  6. 10 Aug, 2022 1 commit
    • Matt's avatar
      TF Examples Rewrite (#18451) · 6eb51450
      Matt authored
      
      
      * Finished QA example
      
      * Dodge a merge conflict
      
      * Update text classification and LM examples
      
      * Update NER example
      
      * New Keras metrics WIP, fix NER example
      
      * Update NER example
      
      * Update MC, summarization and translation examples
      
      * Add XLA warnings when shapes are variable
      
      * Make sure batch_size is consistently scaled by num_replicas
      
      * Add PushToHubCallback to all models
      
      * Add docs links for KerasMetricCallback
      
      * Add docs links for prepare_tf_dataset and jit_compile
      
      * Correct inferred model names
      
      * Don't assume the dataset has 'lang'
      
      * Don't assume the dataset has 'lang'
      
      * Write metrics in text classification
      
      * Add 'framework' to TrainingArguments and TFTrainingArguments
      
      * Export metrics in all examples and add tests
      
      * Fix training args for Flax
      
      * Update command line args for translation test
      
      * make fixup
      
      * Fix accidentally running other tests in fp16
      
      * Remove do_train/do_eval from run_clm.py
      
      * Remove do_train/do_eval from run_mlm.py
      
      * Add tensorflow tests to circleci
      
      * Fix circleci
      
      * Update examples/tensorflow/language-modeling/run_mlm.py
      Co-authored-by: default avatarJoao Gante <joaofranciscocardosogante@gmail.com>
      
      * Update examples/tensorflow/test_tensorflow_examples.py
      Co-authored-by: default avatarJoao Gante <joaofranciscocardosogante@gmail.com>
      
      * Update examples/tensorflow/translation/run_translation.py
      Co-authored-by: default avatarJoao Gante <joaofranciscocardosogante@gmail.com>
      
      * Update examples/tensorflow/token-classification/run_ner.py
      Co-authored-by: default avatarJoao Gante <joaofranciscocardosogante@gmail.com>
      
      * Fix save path for tests
      
      * Fix some model card kwargs
      
      * Explain the magical -1000
      
      * Actually enable tests this time
      
      * Skip text classification PR until we fix shape inference
      
      * make fixup
      Co-authored-by: default avatarJoao Gante <joaofranciscocardosogante@gmail.com>
      6eb51450
  7. 11 Apr, 2022 1 commit
  8. 23 Feb, 2022 1 commit
  9. 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
  10. 12 Dec, 2021 1 commit
  11. 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