Nate Cibik df475bf8e6 Trainer - add cache clearing and the option for batched eval metrics computation (#28769) ... * Added cache clearing for GPU efficiency. * Added cache clearing for GPU efficiency. * Added batch_eval_metrics capability * Ran make fixup * Fixed bug * Fixed whitespace issue * Fixed outdated condition * Updated docstrings with instructions for batch_eval_metrics. Updated end of dataloader logic * Added first version of batch_eval_metrics Trainer test * Fixed batch_eval_metrics Trainer tests for both eval and predict * Fixed batch_eval_metrics behavior for new Trainer variables * Fixed batch_eval_metrics Trainer tests * Ran fixup		2024-05-06 08:23:40 -04:00
..
benchmark
bettertransformer
deepspeed	🚨🚨🚨Deprecate evaluation_strategy to eval_strategy🚨🚨🚨 (#30190)	2024-04-18 12:49:43 -04:00
extended	🚨🚨🚨Deprecate evaluation_strategy to eval_strategy🚨🚨🚨 (#30190)	2024-04-18 12:49:43 -04:00
fixtures	Implementation of SuperPoint and AutoModelForKeypointDetection (#28966)	2024-03-19 14:43:02 +00:00
fsdp	Add FSDP config for CPU RAM efficient loading through accelerate (#30002)	2024-04-22 13:15:28 +01:00
generation	Fix on "cache position" for assisted generation (#30068)	2024-04-23 16:23:36 +05:00
models	[CI update] Try to use dockers and no cache (#29202)	2024-05-06 10:10:32 +02:00
optimization	Add WSD scheduler (#30231)	2024-04-25 12:07:21 +01:00
peft_integration
pipelines	Add chat templating support for KeyDataset in text-generation pipeline (#30558)	2024-04-30 19:51:41 +01:00
quantization	Quantization / HQQ: Fix HQQ tests on our runner (#30668)	2024-05-06 11:33:52 +02:00
repo_utils
sagemaker
tokenization	Remove static pretrained maps from the library's internals (#29112)	2024-03-25 10:33:38 +01:00
tools
trainer	Trainer - add cache clearing and the option for batched eval metrics computation (#28769)	2024-05-06 08:23:40 -04:00
utils	load_image - decode b64encode and encodebytes strings (#30192)	2024-04-26 18:21:47 +01:00
__init__.py
test_backbone_common.py
test_cache_utils.py
test_configuration_common.py
test_configuration_utils.py
test_feature_extraction_common.py
test_feature_extraction_utils.py
test_image_processing_common.py
test_image_processing_utils.py
test_image_transforms.py
test_modeling_common.py	Fix for Neuron (#30259)	2024-05-02 10:24:47 +02:00
test_modeling_flax_common.py	fix: Replace deprecated assertEquals with assertEqual (#30241)	2024-04-15 09:36:06 +01:00
test_modeling_flax_utils.py
test_modeling_tf_common.py
test_modeling_tf_utils.py	Cast bfloat16 to float32 for Numpy conversions (#29755)	2024-03-21 14:04:11 +00:00
test_modeling_utils.py	Update dtype_byte_size to handle torch.float8_e4m3fn/float8_e5m2 types (#30488)	2024-04-26 11:26:43 +01:00
test_pipeline_mixin.py
test_processing_common.py
test_sequence_feature_extraction_common.py
test_tokenization_common.py	fix: Replace deprecated assertEquals with assertEqual (#30241)	2024-04-15 09:36:06 +01:00
test_tokenization_utils.py