CUDA out of memory when using Trainer with compute_metrics

Yes, I solved the “CUDA out of memory when using Trainer with comput_metrics” by using your solution. Thank you so much. :smiling_face_with_three_hearts:

4 Likes