CUDA out of memory when using Trainer with compute_metrics

Thank you! Fixed my problem.