Fix loss
Fix loss

Please review @thomwolf but i think this is equivqlent (and it mimics the loss computation of the original loss)

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