依赖环境

• Python >= 3.6
• torch >= 1.1
• six >= 1.12.0
• argparse
• packaging
• 如果使用混合精度，需要安装英伟达的apex
• 如果涉及到TensorFlow模型的转换，需要安装TensorFlow
• 如果在tokenizer中使用sentencepiece模型，需要安装SentencePiece
• 如果使用模型集成stacking，需要安装LightGBM和BayesianOptimization
• 如果使用全词遮罩（whole word masking）预训练，需要安装分词工具，例如jieba
• 如果使用gap sentence generation预训练目标任务和rouge句子选择策略，需要安装rouge