Skip to content

依赖环境

zhezhaoa edited this page Nov 20, 2022 · 2 revisions
  • Python >= 3.6
  • torch >= 1.1
  • six >= 1.12.0
  • argparse
  • packaging
  • regex
  • 如果使用混合精度,需要安装英伟达的apex
  • 如果涉及到TensorFlow模型的转换,需要安装TensorFlow
  • 如果在tokenizer中使用sentencepiece模型,需要安装SentencePiece
  • 如果使用模型集成stacking,需要安装LightGBM和BayesianOptimization
  • 如果使用全词遮罩(whole word masking)预训练,需要安装分词工具,例如jieba
  • 如果在序列标注下游任务中使用CRF,需要安装pytorch-crf
  • 如果使用超大模型,需要安装DeepSpeed
  • 如果涉及图像模型,需要安装torchvision >= 0.9.0 (torch >= 1.8.0)
  • 如果涉及语音模型,需要安装torchaudio
Clone this wiki locally