Hi, this looks interesting! Can you share more about where exactly you use this function during the training process? For example, with reference to this file: transformers/run_summarization.py at master · huggingface/transformers · GitHub 35
Thank you! @bigheiniu