transformers/tests
Anton Vlasjuk a01b033cb4
Fix galore lr display with schedulers (#31710)
* fix galore lr display with lr schedulers

* style

* add some tests to check for displayed lrs

* copy-paste err for warmup steps

* standardize the default lr to be only in the optimizer

* trying out my luck with the reads
2024-07-05 18:59:09 +01:00
..
agents Code agent: allow function persistence between steps (#31769) 2024-07-05 11:09:11 +02:00
benchmark
bettertransformer
deepspeed Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
extended Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
fixtures
fsdp Llama et al. / FSDP : Fix breaking change in 4.40 for FSDP (#31161) 2024-06-26 14:50:08 +01:00
generation fix assisted decoding (#31401) 2024-07-03 09:22:56 +01:00
models Add training support for SigLIP (#31495) 2024-07-05 14:50:39 +01:00
optimization
peft_integration
pipelines Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
quantization Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
repo_utils
sagemaker Pass datasets trust_remote_code (#31406) 2024-06-17 17:29:13 +01:00
tokenization Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
trainer Fix galore lr display with schedulers (#31710) 2024-07-05 18:59:09 +01:00
utils Fix serialization for offloaded model (#31727) 2024-07-05 08:07:07 +02:00
__init__.py
test_backbone_common.py
test_configuration_common.py Move some test files (tets/test_xxx_utils.py) to tests/utils (#31730) 2024-07-02 13:46:03 +02:00
test_feature_extraction_common.py
test_image_processing_common.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
test_image_transforms.py
test_modeling_common.py Move some test files (tets/test_xxx_utils.py) to tests/utils (#31730) 2024-07-02 13:46:03 +02:00
test_modeling_flax_common.py
test_modeling_tf_common.py
test_pipeline_mixin.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_processing_common.py add initial design for uniform processors + align model (#31197) 2024-06-13 16:27:16 +02:00
test_sequence_feature_extraction_common.py
test_tokenization_common.py Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00