发布时间:2023-04-19 文章分类:电脑百科 投稿人:樱花 字号: 默认 | | 超大 打印

How to change huggingface transformers default cache directory

    • 前言
    • 关于windows上如何修改huggingface transformers默认缓存文件夹

前言

最近在学习用TensorFlow框架做NLP任务,注意到huggingface中的transforms库非常强大,于是开始学习用它来做相应的任务。刚开始用这个库没多久,感觉确实操作起来既简单又强大,于是打算深入学习。

关于windows上如何修改huggingface transformers默认缓存文件夹

第二种方式:在调用from_pretrained函数时使用cache_dir入参,指定缓存文件夹名
如指定缓存位置为当前目录:

AutoModel.from_pretrained('bert-base-chinese', cache_dir='./')

以上