德帕夫洛夫·埃尔莫也是

2024-09-28 01:32:30 发布

您现在位置:Python中文网/ 问答频道 /正文

我尝试使用DeepPavlov库中的ELMO嵌入(ElmoEmbedder)。它的工作非常慢,每100个句子64秒。在

我试图增加mini_batch_的大小,但它并没有加快算法的速度。在

有没有可能加快艾尔莫姆贝德的速度?在


Tags: 算法batch速度句子minielmodeeppavlovelmoembedder
1条回答
网友
1楼 · 发布于 2024-09-28 01:32:30

mini_batch_size只有在GPU上运行模型时,才会对性能产生显著影响。在

如果没有GPU,请尝试使用Google Colab。它为您提供对GPU和TPU的免费访问。在

!pip install deeppavlov

from deeppavlov.deep import find_config
from deeppavlov.core.commands.infer import build_model
config_path = find_config('elmo_ru-wiki')
model = build_model(config_path, load_trained = True, download = True)
a = model(data)

不要忘记在Edit->Notebook settings->Hardware accelerator中启用GPU

相关问题 更多 >

    热门问题