python - 如何更改 .flair 缓存文件夹位置 (https://github.com/flairNLP/flair)？

Question

这个问题是关于 NLP Python 模块 Flair ( https://github.com/flairNLP/flair )。

模型的默认下载文件夹位于 fair 缓存 (~/.flair) 中。但是，在服务器上工作时，我希望在另一个位置进行此下载，因为 /home 目录非常小。这是一个最小的工作示例。

from flair.data import Sentence
from flair.models import SequenceTagger

# make a sentence
sentence = Sentence('I love Berlin .')

# load the NER tagger
tagger = SequenceTagger.load('ner')

因此，您可以看到设备上没有剩余空间，并出现以下错误： (...) OSError: [Errno 28] No space left on device: '~/.flair/models/ner-english/tmp8js3y34i' (...)

Answer 1

发布答案以防有人来到这里并像我一样被卡住。

from pathlib import Path
import flair
#flair.cache_root = "/your/path/.flair" # DOES NOT WORK
flair.cache_root = Path("/your/path/.flair")# WORKS

Answer 2

您可以设置环境变量FLAIR_CACHE_ROOT。通过查看根init .py文件找到。

在某些旧版本的 FlairNLP 中，这不起作用。os.getenv 函数应该传入Path：

cache_root = Path(os.getenv('FLAIR_CACHE_ROOT', Path(Path.home(), ".flair")))