Created by: kolinwei
https://github.com/PaddlePaddle/models/blob/8ea4299422257126605c0ee0989e955bfdc6b818/fluid/PaddleNLP/language_model/gru/train.py#L151
应该改为gpu_num = get_cards(args)