Recently, I want to fine-tuning Bart-base with Transformers (version 4.1.1). The fine-tuning process is very smooth with compute_metrics=None in Trainer. However, when I implement a function of computing metrics and offer this function to Trainer, I received the CUDA out of memory error during the e…

CUDA out of memory when using Trainer with compute_metrics

NicoHambauer February 15, 2022, 12:26pm 6

Thank you for this answer! It helped me a lot!

Topic		Replies	Views
CUDA Out Of Memory when training a DETR Object detection model with compute_metrics 🤗Transformers	4	164	December 3, 2025
Transformer Trainer no response when evaluate with compute_metrics 🤗Transformers	1	203	September 12, 2024
Adding compute_metrics produces Cuda OutOfMemoryError Beginners	0	142	May 22, 2024
Evaluation became slower and slower during Trainer.train() Beginners	8	4771	February 3, 2025
Compute metrics causes OOM Beginners	2	288	February 3, 2024