Sean Doyle

Hey zlw1747832053, I also encountered this error, but found [this](https://discuss.huggingface.co/t/cuda-out-of-memory-when-using-trainer-with-compute-metrics/2941/13) thread to be helpful. It appears that there is some memory leakage when ```transformer.Trainer``` computes metrics on the logits tensor...