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. 06 Feb, 2023 1 commit
    • Sylvain Gugger's avatar
      Update quality tooling for formatting (#21480) · 6f79d264
      Sylvain Gugger authored
      * Result of black 23.1
      
      * Update target to Python 3.7
      
      * Switch flake8 to ruff
      
      * Configure isort
      
      * Configure isort
      
      * Apply isort with line limit
      
      * Put the right black version
      
      * adapt black in check copies
      
      * Fix copies
      6f79d264
  3. 31 Mar, 2021 1 commit
  4. 30 Mar, 2021 1 commit
    • Philipp Schmid's avatar
      Sagemaker test (#10925) · 604c0850
      Philipp Schmid authored
      * init
      
      * first working test
      
      * added todo for setup.py
      
      * working test for single node multi node ddp and smd
      
      * added tensorflow single node test
      
      * added directory for pytorch and tensorflow due to different requirements.txt
      
      * added directory for pytorch and tensorflow
      
      * added comment for run_glue until it is available
      
      * added output_dir to it
      
      * smaller dataset to make test running faster
      
      * adjust HP and script
      
      * adjusted parameter for tensorflow
      
      * refactored test scripts
      
      * adjusted make file
      
      * init
      
      * first working test
      
      * added todo for setup.py
      
      * working test for single node multi node ddp and smd
      
      * added tensorflow single node test
      
      * added directory for pytorch and tensorflow due to different requirements.txt
      
      * added directory for pytorch and tensorflow
      
      * added comment for run_glue until it is available
      
      * added output_dir to it
      
      * smaller dataset to make test running faster
      
      * adjust HP and script
      
      * adjusted parameter for tensorflow
      
      * refactored test scripts
      
      * adjusted make file
      
      * updated dlc container
      
      * commented in all tests
      
      * added both ecr images
      
      * added new master branches
      
      * debug
      
      * added new datasets version
      
      * init
      
      * strange rebase bug
      
      * removed changes
      
      * changed min version for tests to work
      
      * updated DLC
      
      * added model parallel test
      
      * removed test files
      
      * removed test files
      
      * tested with ned dlc
      
      * added correct sagemaker sdk version
      
      * adjust DLCs for official one
      
      * reworked tests
      
      * quality
      
      * removed default profile added documentation to it
      
      * added step in release for sagemaker tests
      
      * reverted version for example script removed duplicated script and added install from master to requirements.txt
      
      * removed mistaken .DS_Stores from mac
      
      * fixed tests
      
      * added Sylvains feedback
      
      * make style
      
      * added lysandre's feedback
      604c0850