Fix example of custom Trainer to reflect signature of compute_loss (#10537)
Fix example of custom Trainer to reflect signature of compute_loss (#10537)


huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/pytorch-transformers,huggingface/transformers