Add and change implementation for RACE dataset

[chengchingwen]

I make the implementation for the RACE dataset. And in order to make the code can run both on RACE and ROCstories, I make some change to the origin implementation inside train.py.