当前位置:   article > 正文

开源python-向transformers学习_python transformer github

python transformer github

python开源系列文章

如何去阅读并学习一些优秀的开源框架的源码?

向transformers学习

https://github.com/huggingface/transformers

功能:支持NLP BERT时代的各种任务与模型

结构

测试

tensorflow的模型为例

  • 测试TF model

    • 准备工作
  • UtilsFunctions测试

    • 测试无网链接时,使用cached file
    • 测试从文件中加载
    • 测试从url加载预训练模型
    • 测试unpack输入
    • 测试xla,softmax在cpu是否稳定
    • 测试从hub中checkpoint分区
    • 测试checkpiunt 分区
  • 检查上传到hub

    • push to hub
    • push to hub中的组织

详细

以AutoModel为中心,加上不同的head用于不同任务

  • ForCausalLM (解码器加上语言模型head,可用于文本生成)
  • ForSequenceClassification (解码器加上分类head,可用于文本分类)

config

参考

  • https://zhuanlan.zhihu.com/p/64770086
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/码创造者/article/detail/887034
推荐阅读
相关标签
  

闽ICP备14008679号