CUDA out of memory when using Trainer with compute_metrics

AndyReas November 14, 2022, 4:22pm 15

What is the reason for only using the first element of logits and predictions?

1 Like

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